I just tried, still the same thing happens. The problem I see is:
In order to be able to call
dual_encoder.load_state_dict(torch.load(‘SAVED_MODEL.pt’))
dual_encoder.eval()
dual_encoder.training == False (or True?)
I first need to import the model file. Running the import command alone will start the training (while not importing some variables!)…