Output channel problem

[eedavidwu]

I noticed that the output or the model has two channels (in paper and code).
But is not clear how to use this two channels results. like(bath_size,2,112,112) into the result (10,1,112,112)