ModuleNotFoundError: No module named 'vgg'

[ydwang-good]

Traceback (most recent call last):
File "/WdHeDisk/users/wangyd/wangyd/24BottleNet/BottleNet++_VGG16.py", line 96, in
model = BottleNetPlusPlus_VGG().to(device)
File "/WdHeDisk/users/wangyd/wangyd/24BottleNet/BottleNet++_VGG16.py", line 59, in init
self.vgg_model = torch.load('vgg16_74.02.pth')
File "/mnt/miniconda3/envs/wangyd/lib/python3.6/site-packages/torch/serialization.py", line 595, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/mnt/miniconda3/envs/wangyd/lib/python3.6/site-packages/torch/serialization.py", line 774, in _legacy_load
result = unpickler.load()
ModuleNotFoundError: No module named 'vgg'

[marina-neseem]

same issue

[ydwang-good]

same issue
it needs to train vgg to get new .pth ourselves