-
Notifications
You must be signed in to change notification settings - Fork 273
Open
Description
Hi, I started to learn pytorch a few months ago,
so I might be questioning basic stuff.
in [Non-local_pytorch/demo_MNIST_train.py](url) file,
there was net.train()
in the 50th line, within the training loop.
after the testing loop, net.train()
is called once more in the 70th line.
I know that switching the network into training mode after the testing loop is important,
but why is net.train()
in the 70th line necessary when it would be called at the beginning of the training loop anyways?
Metadata
Metadata
Assignees
Labels
No labels