Hi all,
For some purpose, I want to use the eval() mode for BatchNorm layers and train() mode for Dropout layers during training. Does anyone know how to do this? If I simply call model.train() or model.eval(), both of them will be in train mode or evaluation mode.