How to check if a model is in train or eval mode in Pytorch?

Question

How to check from within a model if it is currently in train or eval mode?

score 27 · Accepted Answer · answered Dec 17 '20 at 16:27

27

From the Pytorch forum, with a small tweak:

use

if self.training:
    # it's in train mode
else:
    # it's in eval mode

Always better to have a stack overflow answer than to look at forums.

answered Dec 17 '20 at 16:27

Gulzar

1 Answers1