赞
踩
module的内部包含一个training的属性(bool类型),调用training即可。
if net.training xxx else: yyy
多说一句:一般我们调用net.eval() 或者net.train()都会对这个training属性进行改写。