Hello! I faced a problem when I was using torch.load(args.checkpoint) to load the pre-trained model (its called model.pth when I downloaded it, not the food2k_resnet50_0.0001.pth):
AttributeError: Can't get attribute 'PMG' on <module 'model' from '/nas/users/prenet/model.py'>
But I cannot find any attribute called 'PMG' in model.py or other scripts, any idea about this error? Thank you.
Hello! I faced a problem when I was using
torch.load(args.checkpoint)to load the pre-trained model (its called model.pth when I downloaded it, not the food2k_resnet50_0.0001.pth):But I cannot find any attribute called 'PMG' in model.py or other scripts, any idea about this error? Thank you.