search

nikhilroxtomar / FANet

FANet: A Feedback Attention Network for Improved Biomedical Image Segmentation [IEEE Transactions on Neural Networks and Learning Systems]
https://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=9741842
30 stars 8 forks source link

Input output mask does not correspond. #1

Closed lancesye closed 2 years ago

lancesye commented 2 years ago

Shuffle set True in train_loader. In a new training epoch, the orders of training sample will suffle, not same as the last return_train_mask.

nikhilroxtomar commented 2 years ago

Thanks for valuable feedback, I have updated the code and set shuffle=False. The code would not work properly with the shuffle set as True.