Help wanted
Chris36839 opened this issue · 2 comments
Hi, I'm encountering the following error while running "Train_MindEye.py" and am unable to resolve it. I hope someone can help me identify the issue and suggest possible solutions.
Firstly, I got the following error:
Traceback (most recent call last):
File "/mnt/d/Files/Codes/fMRI-reconstruction-NSD/src/Train_MindEye.py", line 585, in
for train_i, (voxel, image, coco) in enumerate(train_dl):
File "/root/miniconda3/envs/mindeye/lib/python3.10/site-packages/accelerate/data_loader.py", line 630, in iter
self.set_epoch(self.iteration)
File "/root/miniconda3/envs/mindeye/lib/python3.10/site-packages/accelerate/data_loader.py", line 707, in set_epoch
if hasattr(self.batch_sampler.sampler, "set_epoch"):
AttributeError: 'NoneType' object has no attribute 'sampler'
So, I changed "utilis.py"
train_dl = torch.utils.data.DataLoader(train_data, batch_size=32, num_workers=1, shuffle=False), set batch_size to 32
but I still can not run the training process
Hey,have you resolved this problem?