MedARC-AI / fMRI-reconstruction-NSD

fMRI-to-image reconstruction on the NSD dataset.
MIT License
294 stars 39 forks source link

Error when using --no-hidden flag #35

Closed MSurfer20 closed 10 months ago

MSurfer20 commented 10 months ago

If we run the script with the hidden arg set to False, the following error is encountered

Traceback (most recent call last):
  File "/home2/suyash.mathur/Brain_project/fMRI-reconstruction-NSD/src/Train_MindEy
e.py", line 604, in <module>
    clip_voxels, clip_voxels_proj = diffusion_prior.module.voxel2clip(voxel) if distri
buted else diffusion_prior.voxel2clip(voxel)
ValueError: not enough values to unpack (expected 2, got 1)
Traceback (most recent call last):
  File "/home2/suyash.mathur/Brain_project/fMRI-reconstruction-NSD/src/Train_MindEy
e.py", line 604, in <module>
    clip_voxels, clip_voxels_proj = diffusion_prior.module.voxel2clip(voxel) if distri
buted else diffusion_prior.voxel2clip(voxel)
ValueError: not enough values to unpack (expected 2, got 1)
Traceback (most recent call last):
  File "/home2/suyash.mathur/Brain_project/fMRI-reconstruction-NSD/src/Train_MindEy
e.py", line 604, in <module>
    clip_voxels, clip_voxels_proj = diffusion_prior.module.voxel2clip(voxel) if distri
buted else diffusion_prior.voxel2clip(voxel)
ValueError: not enough values to unpack (expected 2, got 1)
ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0
 (pid: 2575) of binary: /home2/suyash.mathur/anaconda3/envs/mindeye/bin/python
Traceback (most recent call last):
  File "/home2/suyash.mathur/anaconda3/envs/mindeye/bin/accelerate", line 8, in <modul
e>
    sys.exit(main())
  File "/home2/suyash.mathur/anaconda3/envs/mindeye/lib/python3.10/site-packages/accel
erate/commands/accelerate_cli.py", line 47, in main
    args.func(args)
  File "/home2/suyash.mathur/anaconda3/envs/mindeye/lib/python3.10/site-packages/accel
erate/commands/launch.py", line 977, in launch_command
    multi_gpu_launcher(args)
  File "/home2/suyash.mathur/anaconda3/envs/mindeye/lib/python3.10/site-packages/accel
erate/commands/launch.py", line 646, in multi_gpu_launcher
    distrib_run.run(args)
  File "/home2/suyash.mathur/anaconda3/envs/mindeye/lib/python3.10/site-packages/torch
/distributed/run.py", line 753, in run
[
MSurfer20 commented 10 months ago

Nvm, figured out wrong usage of use_projector argument.