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
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