DQN算法优化器迁移，loss函数的回调问题

问题描述：

【功能模块】

想把一个基于pytorch的DQN玩flappy bird项目转为mindspore框架

在torch项目中,loss回传如下图

 

参考优化器迁移指南 优化器迁移指南 — MindSpore master documentation

 

改写成下面形式

 

在测试后，发现实际上并没有进行学习。

【操作步骤&问题现象】

1、在改写的封装Model函数中，并没有使用到y_predict 和 y_batch_tensor 生成的loss值，请问要如何修改放入Model函数中，

2、在改写的形式中，是否loss函数并没有回传，应该如何修改才能传回？

具体网络如下

解答：

您好，model init里会把 net loss opt封装成一个大的cell，具体可以参考 withlosscell、trainonestepcell的定义