小行星

热爱平淡,向往未知

  • Home
  • Categories
  • Archives
  • About
  • top
  • Search
  • Table of Contents
  • Overview

fangyh

最爱的是那苍穹之外的浩渺宇宙
597 posts
4 categories
66 tags
0%

pytorch retain_graph=True

Posted on 2018-08-Mon Views: Views:

pytorch 在backward的时候会删除计算图,
所以在backward的时候要加入参数 loss.backward(retain_graph=True)

Loss.backward() raises error ‘grad can be implicitly created only for scalar outputs’
梯度为0
© 2022 fangyh
Powered by Hexo & NexT.Muse