CyclicLRWithRestarts 的第一批数据被双计数,以打破代码
作者: sergiosolorzano创建于 2024年11月11日更新于 2024年11月11日
你好,我认为通过遵循 调度器在训练循环中的建议位置,第一个批次被计数了两次:首先在顶部(因为 scheduler.step() 调用了 batch_step(),batch_step() 会增加迭代),然后再次在 scheduler.batch_step() 中。这导致了 batch_step() 中的索引越界 t_cur = self.t_epoch + self.batch_increments[self.iteration]
内容来源: sebastianstarke/AI4Animation