more training

cvae_pyramid.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32_skin import config
+from configs.iccv.config_abl_pyr_32 import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -794,7 +794,7 @@ if __name__ == '__main__':
     # generate_results()
 
     ckpt = tf.train.Checkpoint(model=model)
-    manager = tf.train.CheckpointManager(ckpt, 'checkpoints/post/' + config['name'], max_to_keep=20)
+    manager = tf.train.CheckpointManager(ckpt, 'checkpoints/iccv/' + config['name'], max_to_keep=20)
     train_accumulator = Accumulator()
     valid_accumulator = Accumulator()
 
@@ -854,7 +854,7 @@ if __name__ == '__main__':
         train_loss = train_step(batch)
         train_accumulator.update(train_loss)
 
-        if step % 50 == 0:
+        if step % 25 == 0:
             data_valid = valid_iterator.get_next()
             data_valid = static_dataset_valid.expand_batch(batch_to_dict(data_valid, config['signature_dynamic']))
             valid_loss = valid_step(data_valid)
@@ -882,7 +882,7 @@ if __name__ == '__main__':
 
         if step % 5000 == 0:
             manager.save(checkpoint_number=step)
-            
+
         print(step)
 
         step += 1