目前正在尝试迁移TF2的脚本,迁移过程中不知道如下部分如何合理地进行转换。尤其是不希望大量改动原来的代码,借助GradOperation算子是否可以实现红框中根据loss计算gradient的功能?
mindspore的求导是:
GradOperation()(net)(*inputs)
所以需要把输出loss值的网络提取出来,并且在这个网络上使用GradOperation进行求导。
目前正在尝试迁移TF2的脚本,迁移过程中不知道如下部分如何合理地进行转换。尤其是不希望大量改动原来的代码,借助GradOperation算子是否可以实现红框中根据loss计算gradient的功能?
mindspore的求导是:
GradOperation()(net)(*inputs)
所以需要把输出loss值的网络提取出来,并且在这个网络上使用GradOperation进行求导。