prediction_gan:PyTorch Impl。 预测优化器(以稳定GAN训练)

时间:2024-05-18 22:56:02
【文件属性】:

文件名称:prediction_gan:PyTorch Impl。 预测优化器(以稳定GAN训练)

文件大小:9KB

文件格式:ZIP

更新时间:2024-05-18 22:56:02

optimizer pytorch gan dcgan stabilizing-adversarial-nets

预测优化器(稳定GAN训练) 介绍 这是以下论文中介绍的“预测方法”的PyTorch实现... Abhay Yadav等人,《使用预测方法稳定对抗网络》,ICLR 2018, (为澄清起见,我不是本文的作者。 ) 作者提出了一种简单(但有效)的方法来稳定GAN训练。 借助此Prediction Optimizer,您可以轻松地将该方法应用于现有的GAN代码。 这暗示。 与大多数PyTorch优化器和网络结构兼容。 (如果有任何问题,请告诉我) 如何使用 指示 导入预测.py from prediction import PredOpt 像优化程序一样初始化pred = PredOpt(net.parameters()) 在“ with”块中运行模型,以从具有预测参数的模型中获取结果。 使用'step'参数,您可以控制提前步长(默认为1.0) with pred . lookah


【文件预览】:
prediction_gan-master
----example_dcgan.py(11KB)
----tests()
--------test_context.py(3KB)
----run_example.sh(275B)
----.gitignore(1KB)
----celeba_data_preprocess.py(966B)
----README.md(5KB)
----prediction.py(2KB)

网友评论