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

snapshot...

parent 2488fbb7
Branches
No related tags found
No related merge requests found
...@@ -66,7 +66,6 @@ class Trainer(object): ...@@ -66,7 +66,6 @@ class Trainer(object):
decay_step = decay_params["step"] decay_step = decay_params["step"]
decay_factor = decay_params["factor"] decay_factor = decay_params["factor"]
total_steps = phase_args["num_steps"] total_steps = phase_args["num_steps"]
print('total number of steps: ', total_steps)
metric = tf.keras.metrics.Mean() metric = tf.keras.metrics.Mean()
psnr_metric = tf.keras.metrics.Mean() psnr_metric = tf.keras.metrics.Mean()
...@@ -79,6 +78,7 @@ class Trainer(object): ...@@ -79,6 +78,7 @@ class Trainer(object):
checkpoint = tf.train.Checkpoint( checkpoint = tf.train.Checkpoint(
G=generator, G=generator,
G_optimizer=G_optimizer) G_optimizer=G_optimizer)
ckpt_manager = tf.train.CheckpointManager(checkpoint, self.model_dir, max_to_keep=7)
status = utils.load_checkpoint(checkpoint, "phase_1", self.model_dir) status = utils.load_checkpoint(checkpoint, "phase_1", self.model_dir)
logging.debug("phase_1 status object: {}".format(status)) logging.debug("phase_1 status object: {}".format(status))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment