You could call either
- call
.train()
on the rnn module after usingmodel.eval()
- call
.eval()
only on the necessary modules - or disable cudnn via
torch.backends.cudnn.enabled = False
You could call either
.train()
on the rnn module after using model.eval()
.eval()
only on the necessary modulestorch.backends.cudnn.enabled = False