Assign loss calculated outside of graph to Pytorch model

后端 未结 0 1501
予麋鹿
予麋鹿 2020-12-21 03:19

I need to find a way to use a non-pytorch loss function on a pytorch model. For reasons I will not go into it would be extremely cumbersome to rewrite this loss function in

相关标签:
回答
  • 消灭零回复
提交回复
热议问题