3D训练时loss的梯度更新和参数更新

[Mhaiyang]

您好，

想问下，目前代码中3d训练部分，是整个video计算平均loss然后梯度回传更新参数。是否可以每一帧计算loss进行梯度回传并更新参数，这样是否就可以把video的length从2变为整个video的长度？ 涉及到memory相关的代码，应该如何修改呢？

期待并感谢您的回复！