Skip to content
Snippets Groups Projects
Commit 2ba2abeb authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 556b9a69
No related branches found
No related tags found
No related merge requests found
import tensorflow as tf
import tensorflow_addons as tfa
#import tensorflow_addons as tfa
from util.setup import logdir, modeldir, cachepath
from util.util import homedir
import subprocess
import os, datetime
import numpy as np
......@@ -401,7 +400,7 @@ class IcingIntensityNN:
self.learningRateSchedule = tf.keras.optimizers.schedules.ExponentialDecay(initial_learning_rate, decay_steps, decay_rate)
optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule)
optimizer = tfa.optimizers.MovingAverage(optimizer)
# optimizer = tfa.optimizers.MovingAverage(optimizer)
if TRACK_MOVING_AVERAGE:
ema = tf.train.ExponentialMovingAverage(decay=0.999)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment