其他分享
首页 > 其他分享> > pytorch中一些常见函数的用法

pytorch中一些常见函数的用法

作者:互联网

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