search

Zj-BinXia / SSL

This project is the official implementation of 'Structured Sparsity Learning for Efficient Video Super-Resolution', CVPR2023
95 stars 6 forks source link

Can not test SSL_Vimeo_BD.pt #13

Open House-Leo opened 8 months ago

House-Leo commented 8 months ago

model.load_prune(args,opt) [0/1920] File "/data1/project/SSL/basicsr/models/video_recurrent_prune_model.py", line 245, in load_prune prune_info = torch.load(load_path) File "/home/miniconda3/envs/SSL/lib/python3.8/site-packages/torch/serialization.py", line 789, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "/home//miniconda3/envs/SSL/lib/python3.8/site-packages/torch/serialization.py", line 1131, in _load result = unpickler.load() File "/home/miniconda3/envs/SSL/lib/python3.8/site-packages/torch/nn/parallel/distributed.py", line 791, in __setstate__ self._build_replicated_tensor_module() File "/home/miniconda3/envs/SSL/lib/python3.8/site-packages/torch/nn/parallel/distributed.py", line 676, in _build_replicated_tensor_module if self._use_replicated_tensor_module: File "/home/miniconda3/envs/SSL/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1265, in __getattr__ raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'DistributedDataParallel' object has no attribute '_use_replicated_tensor_module'

The SSL_Vimeo_BI.pt is work, but BD pre-trained model has bug.

Zj-BinXia commented 8 months ago

You can try to run "bash ./scripts/dist_test_prune_l1.sh 1 options/test/BasicVSR/test_BasicVSR_Vimeo90K_BDx4.yml" rather than "CUDA_VISIBLE_DEVICES=1 bash ./scripts/dist_test_prune_l1.sh 1 options/test/BasicVSR/test_BasicVSR_Vimeo90K_BDx4.yml"