欲在examples/imagenet文件夹下的main.py中引入pytorch_pretrained_vit文件夹下的model.py
原来的程序中使用的是下面这条代码:
from pytorch_pretrained_vit import ViT, load_pretrained_weights
报错:
ModuleNotFoundError: No module named 'pytorch_pretrained_vit'
解决方法:引入sys库,使用sys.path.append()函数,在pytorch_pretrained_vit后面加上.model
import sys
sys.path.append(r'../..')
from pytorch_pretrained_vit.model import ViT, load_pretrained_weights
sys.path.append()函数是将括号中的路径加入环境变量中,使其可以被调用。