Merge branch 'main' of github.com:TRI-ML/RAP into main

TRI-ML · Mar 14, 2023 · bf2232f · bf2232f
2 parents 04387f9 + ed6881c
commit bf2232f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/risk_biased/config/waymo_config.py b/risk_biased/config/waymo_config.py
@@ -51,9 +51,9 @@
 learning_rate = 3e-4
 batch_size = 64
 accumulate_grad_batches = 2
-num_epochs_cvae = 0
+num_epochs_cvae = 200
 num_epochs_bias = 100
-gpus = [1]
+gpus = [0]
 seed = 0  # Give an integer value to seed will set seed for pseudo-random number generators in: pytorch, numpy, python.random
 num_workers = 8