Simple Tensorflow implementation of Stabilizing Adversarial Nets With Prediction Methods (ICLR 2018)
from Adam_prediction import Adam_Prediction_Optimizer
G_op = Adam_Prediction_Optimizer(learning_rate=0.001, beta1=0.9, beta2=0.999, prediction=True)
D_op = Adam_Prediction_Optimizer(learning_rate=0.001, beta1=0.9, beta2=0.999, prediction=False)
- adam : 7.12
- adam-prediction : 7.17
Junho Kim