skin_gnn

cvae.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_skin_curv import config
+from configs.iccv.config_abl_skin import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -34,7 +34,6 @@ else:
     from losses import edge_loss
 
 
-
 from losses import collision_ratio, kl_loss, normal_loss #, concept_loss
 # from losses import collision_penalty as collision_penalty
 from losses import collision_loss as collision_penalty
@@ -552,7 +551,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)