pytorch自定义loss及其loss_backward()部分

model输出维度是[n_batch,classes,n_patch,n_patch],出model后需要根据一系列运算将它映射为维度是[n_batch,classes]的分类结果,并计算Loss,映射和loss计算想都写在自定义的loss函数中,设计到中间变量的保存,经查阅loss_backward()也要自己设计,loss才能正确传播,求给个范例(有中间变量的保存和loss_back()设计最好)。
已邀请:

要回复问题请先登录注册