pytorch中一些常见函数的用法

model.train():打开BN和Droupout

model.eval():关闭BN和Droupout

loss.backward()：loss的梯度回传

 

optimizer.step()：进行一次参数更新

*表示逐点相乘，@表示矩阵相乘