When model uses specific layers/part-Dropouts Layers, BatchNorm Layers etc, it can be turned on during the train mode, but it needs to be turned off during the evaluation mode. model.train() model.eval() Sets model in training mode: activates specific layers/part-Dropout layers etc normalisation layers1 use per-batch statistics Sets model in evaluation (inference) mode: de-activates Dropout laye..