That does actually you need to define the model first and then do the forward pass like this
model = CNNModel1(fully_layer_1, fully_layer_2, drop_rate)
model.load_state_dict(torch.load(‘VA2.pth’))
# and then the forward pass
output = model(input)

where you pass in the same parameters for your training into CNNModel1