今天在运行TransGAN的时候出现了下面的错误:
File "train_derived.py", line 6, in <module>
import models_search
File "/data/tianhao.lu/code/TransGAN/models_search/__init__.py", line 1, in <module>
from models_search import ViT_custom_local544444_256_rp, ViT_scale3_local_new_rp, diff_aug, ada, ViT_helper, ViT_custom
File "/data/tianhao.lu/code/TransGAN/models_search/ViT_custom_local544444_256_rp.py", line 6, in <module>
from models_search.diff_aug import DiffAugment
File "/data/tianhao.lu/code/TransGAN/models_search/diff_aug.py", line 11, in <module>
from torch_utils import misc
File "/data/tianhao.lu/code/TransGAN/torch_utils/misc.py", line 64, in <module>
symbolic_assert = torch.Assert # 1.7.0
AttributeError: module 'torch' has no attribute 'Assert'
解决方法
pip install torch==1.7.1