model.train()
sets model.training
to True, model.eval()
to False
model
be an object of type inheriting from torch.nn.Module
.
What does model.train()
and model.eval()
do?
Answer
model.train()
sets model.training
to True, model.eval()
to False
Relevant part of lecture