shoutOutYangJie/FaceID-GAN

关于optimizer.step()更新顺序的问题

Opened this issue · 0 comments

楼主你好,我想问一下对于这种既有分类器又有判别器的架构,可不可以先把所有的loss都backward()之后,再一次执行判别器,分类器,生成器等等模块的optimizer.step()
因为我之前自己实现CVAE-GAN的时候,先执行C的zero_gard,再执行loss_C.backward(),然后再执行optimizer_C.step(),按照同样步骤运行到D的backward就会报错,但是先把所有loss都bankward之后再更新参数就没报错
所以这种backward+step的步骤 顺序是绑定的吗 按照我那种方式统一step可不可以- -