Hi, it really confuses me that why the DQN tutorial has no model.eval() and model.train(), since there are several BatchNorm layers in the model.
As I understand, model.eval() should be put before calling select_action(). And in function optimize_model(), model.train() should be put before calling the first model.forward() while model.eval() should be put before calling the second one. Is it correct?