Train error for any dataset asides Tumor

[b3r-prog]

Hello

I am trying to train on the differrent task. While the braintumor Task01 training works, the others dont. I get this error below

File ".../medical-decathlon/2D/dataloader.py", line 204, in generate_batch_from_files img = img[:,:,:,0] # Just take FLAIR channel (channel 0) IndexError: too many indices for array: array is 3-dimensional, but 4 were indexed

Is there an argument to specify or has this issue been noticed before?

@ravi9 @tonyreina