Skip to content

net.train() again in demo_MNIST_train.py file #48

@Dylee-y

Description

@Dylee-y

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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions