rui-yan / SSL-FL

Self-supervised federated learning for medical imaging - IEEE TMI
63 stars 9 forks source link

Update data_utils.py #15

Closed chillum-codeX closed 1 year ago

chillum-codeX commented 1 year ago

fix this error"The expected size is [3, 224, 224], but the actual size for entry 0 is [3, 224, 224], and for entry 5 is [3, 224, 360]." change in line 123 in data_utils.py in getitem method of class DatasetFLFinetune, where the sample tensor is being returned. this changes able to do validation on images