Module.trainModule.eval 方法和 Module.training属性的关系

    xiaoxiao2022-07-02  165

    Module.train()、Module.eval() 方法和 Module.training属性的关系

    print(net.training, net.submodel1.training) net.train() # 将本层及子层的training设定为True net.eval() # 将本层及子层的training设定为False net.training = True # 注意,对module的设置仅仅影响本层,子module不受影响 net.training, net.submodel1.training
    最新回复(0)