search

swz30 / MIRNetv2

[TPAMI 2022] Learning Enriched Features for Fast Image Restoration and Enhancement. Results on Defocus Deblurring, Denoising, Super-resolution, and image enhancement
Other
401 stars 56 forks source link

Unable to validate model results #10

Open home56200 opened 2 years ago

home56200 commented 2 years ago

The model has been trained from the training file, but it cannot be verified in the test.py( \basicsr\ ). The error in the picture below will appear. How can I solve it ??? thank you

image

" This is the wrong part " (psnr_ssim.py) image

Lz-2019317 commented 1 year ago

报错如下： ModuleNotFoundError: No module named 'basicsr' 这个basicsr是项目文件夹，导入为沙会报错阿？ 报错如下： ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 28610) of binary: /home/xtzg/anaconda3/envs/pytorch1.1 我的设置： yaml配置文件中也修改shuffle为false # data loader use_shuffle: false # true num_worker_per_gpu: 0 # 8 batch_size_per_gpu: 1 # 8 train.sh中设置为： python -m torch.distributed.run --nproc_per_node=1 --master_port=4321 basicsr/train.py -opt $CONFIG --launcher pytorch