Update configurations

ika-rwth-aachen · Oct 13, 2022 · e7385f3 · e7385f3
1 parent 6540b07
commit e7385f3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/range_image_compression/configs/additive_lstm_cfg.py b/range_image_compression/configs/additive_lstm_cfg.py
@@ -39,7 +39,7 @@ def additive_lstm_cfg():
     cfg.val_batch_size = 128
     cfg.save_freq = 10000
     cfg.train_output_dir = "output"
-    cfg.xla = False
+    cfg.xla = True
     cfg.mixed_precision = False
 
     # Learning Rate scheduler

diff --git a/range_image_compression/configs/additive_lstm_slim_cfg.py b/range_image_compression/configs/additive_lstm_slim_cfg.py
@@ -35,8 +35,8 @@ def additive_lstm_slim_cfg():
 
     # Training
     cfg.epochs = 3000
-    cfg.batch_size = 256
-    cfg.val_batch_size = 256
+    cfg.batch_size = 128
+    cfg.val_batch_size = 128
     cfg.save_freq = 10000
     cfg.train_output_dir = "/output"
     cfg.xla = True