Yes I mean use the CNNModel1 code in *.py to recreate your model and then you will load your state dict on to that model so like this
class CNNModel1(nn.Module):
code for defining the model here. So you need the init function and forward function with all the code from *.py

model = CNNModel1(PARAMETERS)
model.load_state_dict(torch.load(“VA2.pth”))


where parameters are the ones you used to train your model.