在pytorch中,哪个部分耗时最多?损失计算还是backward+step?

In pytorch, which part cost most time? Loss calculation or backward+step?

在pytorch中,哪个部分花费的时间最多?

  1. 损失计算(没有倒退,只是计算)
  2. 落后
  3. 步 如果可以请解释一下。

不仅在 Pytorch 中,在所有框架中,backward_step 需要更多时间