A bug for the input

In the TransBTS_downsample8x_skipconnection.TransformerBTS:

self.Unet = Unet(in_channels=4, base_channels=16, num_classes=4)

This code would force the input has 4 channel. I think it should be modified as:

self.Unet = Unet(in_channels=num_channels, base_channels=16, num_classes=4)

Am I right?

My point :
for the Brats Dataset, the input does have 4 channel, ie. t1, t1ce, t2, flair,
so I think it's correct, no bug.

For Brats Dataset, it is OK. But, for another dataset, the channel may be not 4. So, self.Unet = Unet(in_channels=num_channels, base_channels=16, num_classes=4) may be more robust.

For Brats Dataset, it is OK. But, for another dataset, the channel may be not 4. So, self.Unet = Unet(in_channels=num_channels, base_channels=16, num_classes=4) may be more robust.

yes， agree with you