Skip to content

The .train(False) or .eval() does not turn off the gradient computing #1756

Closed
@sonack

Description

@sonack

net.train(False) # Don't need to track gradents for validation

and
net.train(True) # Turn gradients back on for training

The comments is wrong maybe?

cc @suraj813 @aaronenyeshi @chaekit @sekyondaMeta @svekars @carljparker @NicolasHug @kit1980 @subramen @robieta

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions