首页 > 其他分享> > pytorch中一些常见函数的用法 pytorch中一些常见函数的用法 2021-08-16 17:01:09 作者:互联网 model.train():打开BN和Droupout model.eval():关闭BN和Droupout loss.backward():loss的梯度回传 optimizer.step():进行一次参数更新 *表示逐点相乘,@表示矩阵相乘 标签:loss,函数,相乘,BN,Droupout,用法,pytorch,回传,model 来源: https://www.cnblogs.com/zukang/p/15148710.html