search

nupurkmr9 / vision-aided-gan

Ensembling Off-the-shelf Models for GAN Training (CVPR 2022 Oral)
https://www.cs.cmu.edu/~vision-aided-gan/
MIT License
379 stars 26 forks source link

An error occurred while resuming training using“ --resume” #7

Closed 49xxy closed 1 year ago

49xxy commented 1 year ago

Thank you for your excellent work. When I wanted to resume training after a break, I encountered the following problems: Constructing networks... Setting up augmentation... Resuming from "/content/drive/MyDrive/vision-aided-gan/training-runs/00000-pv-lianpu7k-paper256_2fmap-kimg5000-batch8-ada-input-clip-output-conv_multi_level-cv_loss_multilevel_sigmoid_s-augcv_ada/network-snapshot-000300.pkl" Traceback (most recent call last): File "train.py", line 658, in <module> main(args.parse_args()) File "train.py", line 650, in main subprocess_fn(rank=0, args=args, temp_dir=temp_dir) File "train.py", line 467, in subprocess_fn training_loop.training_loop(rank=rank, **args) File "/content/drive/MyDrive/vision-aided-gan/training/training_loop.py", line 202, in training_loop resume_data = legacy.load_network_pkl(f) File "/content/drive/MyDrive/vision-aided-gan/legacy.py", line 23, in load_network_pkl data = CPU_Unpickler(f).load() File "/content/drive/MyDrive/vision-aided-gan/legacy.py", line 88, in find_class return super().find_class(module, name) AttributeError: Can't get attribute '_LinearWithBias' on <module 'torch.nn.modules.linear' from '/usr/local/lib/python3.8/site-packages/torch/nn/modules/linear.py'>

I sincerely hope to get your help

nupurkmr9 commented 1 year ago

Hi, Thanks for the interest in or work. Can you comment which PyTorch version you are using that results in this error?

49xxy commented 1 year ago

AttributeError: Can't get attribute '_LinearWithBias' on <module 'torch.nn.modules.linear'

Hi！I use torch1.12.1+cu113.