small parameter updates

ika-rwth-aachen · Sep 28, 2022 · ba4fc1f · ba4fc1f
1 parent 5d2ff78
commit ba4fc1f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/range_image_compression/configs/additive_lstm_demo_cfg.py b/range_image_compression/configs/additive_lstm_demo_cfg.py
@@ -39,12 +39,12 @@ def additive_lstm_demo_cfg():
     cfg.val_batch_size = 1
     cfg.save_freq = 20
     cfg.train_output_dir = "output"
-    cfg.xla = False
+    cfg.xla = True
     cfg.mixed_precision = False
 
     # Learning Rate scheduler
-    cfg.lr_init = 2e-3
-    cfg.min_learning_rate = 2e-8
+    cfg.lr_init = 5e-4
+    cfg.min_learning_rate = 5e-7
     cfg.min_learning_rate_epoch = cfg.epochs
     cfg.max_learning_rate_epoch = 0
 

diff --git a/range_image_compression/utils.py b/range_image_compression/utils.py
@@ -23,7 +23,7 @@
 #  ==============================================================================
 
 import tensorflow as tf
-import numpy as np
+
 
 from architectures import Additive_LSTM_Demo, Additive_LSTM, Additive_LSTM_Slim, Additive_GRU, Oneshot_LSTM
 from configs import additive_lstm_demo_cfg, additive_lstm_cfg, additive_lstm_slim_cfg, additive_gru_cfg, oneshot_lstm_cfg