Closed Ilya-Muromets closed 4 years ago
Thanks @Ilya-Muromets , but optimizer.step() has been called in train_model.train().
optimizer.step()
train_model.train()
Got it, removed.