关于forward函数和自动梯度

Question

之前曾经在__init__中把 训练层 添加在了self下的一个字典里面
self.layer_list[0] = make_layer()
在forward中训练
out = self.layer_list[0]_(x)

发现训练出错了，准确率一直上不去

如果把层直接添加在self下
self.layer1 = make_layer() ,
在forward中训练
out = self.layer1(x)
就可以了

是不是因为梯度计算出问题，因为没有直接添加在layer下面，如果是的话，
在backward里如何根据 forward 里面的步骤，计算出相应的反向梯度呢

Answer 1

会写代码的好厨师 2023-08-24 12:01:04

放在一个字典里和放在变量单独定制，没有什么区别，应该是字典使用的时候的问题，具体要看下上下文的代码才能判断