Open minnieteng opened 3 years ago
Hello! When I try to instantiate the trained model this way:
atType = ['self-attention', 'self_relation-attention'][0] model = networks.resnet18_at(at_type=atType) model.load_state_dict(torch.load('trained_model_debian.pth', map_location=get_default_device()))
I keep getting this error: what do I do??
`--------------------------------------------------------------------------- RuntimeError Traceback (most recent call last)
It loads based on the Diction of Python. I feel that you can check whether the key value is consistent.
Hello! When I try to instantiate the trained model this way:
I keep getting this error: what do I do??
`--------------------------------------------------------------------------- RuntimeError Traceback (most recent call last)