Closed OeslleLucena closed 3 years ago
Hi, what makes you think this is related to RandCropByPosNegLabeld
?
This part of the error:
output = model_step(*args)``
File "/home/ol18/Codes/TractUncertainty/segmentation_sh_torchIO_PIL MONAI.py", line 224, in training_step
print('train',batch.keys())
AttributeError: 'list' object has no attribute 'keys'
implies to me that it's trying to use a dictionary-based transform on a list. I.e., subjects_list[TRAINING]
looks to be a list as opposed to a dictionary. is that correct?
Hi @rijobr! You're right, I just don't understand why this is happening. I swap subjects_list[TRAINING]
for subjects_list[VALIDATION]
to see if the problem was with subjects_list[TRAINING]
but the error is the same.
Any idea what could potentially be?
Running a minimal example for a patch-based segmentation based on the spleen Lightning segmentation tutorial from here. However, the returned training batch is a list instead of a dict. The validation bit works with no error. Here is a print of the error:
Below is shown how the transforms, training/validation steps, and loaders are coded.