relaxed logging to empty directory

jackd · May 9, 2020 · edab69c · edab69c
1 parent df90fed
commit edab69c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow_graphics/projects/pointnet/helpers.py b/tensorflow_graphics/projects/pointnet/helpers.py
@@ -122,7 +122,8 @@ def setup_tensorboard(flags):
     return
 
   # --- Do not allow experiment with same name
-  assert not tf.io.gfile.exists(flags.logdir), \
+  assert (not tf.io.gfile.exists(flags.logdir) or
+          len(tf.io.gfile.listdir(flags.logdir)) == 0), \
     "CRITICAL: folder {} already exists".format(flags.logdir)
 
   # --- Log where summary can be found