Commit a614af3f authored by Kai Arulkumaran's avatar Kai Arulkumaran Committed by Alykhan Tejani
Browse files

Add model.train() + model.eval() to models docs (#405)

parent c84aa998
......@@ -36,6 +36,11 @@ These can be constructed by passing ``pretrained=True``:
densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(pretrained=True)
Some models use modules which have different training and evaluation
behavior, such as batch normalization. To switch between these modes, use
``model.train()`` or ``model.eval()`` as appropriate. See
:meth:`~torch.nn.Module.train` or :meth:`~torch.nn.Module.eval` for details.
All pre-trained models expect input images normalized in the same way,
i.e. mini-batches of 3-channel RGB images of shape (3 x H x W),
where H and W are expected to be at least 224.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment