Merge pull request #30 from Cubevoid/final_data_collection

Final data collection
Cubevoid · Apr 19, 2024 · 19fba51 · 19fba51
2 parents a53de02 + 62bf38f
commit 19fba51
Show file tree

Hide file tree

Showing 232 changed files with 217 additions and 34 deletions.
diff --git a/configs/data_collection/config.yaml b/configs/data_collection/config.yaml
@@ -3,7 +3,7 @@ collector:
   game: Pong
   num_samples: 100
   max_num_objects: 32
-  model: "SAM"
-  # model: "FastSAM-x"
+  # model: "SAM"
+  model: "FastSAM-x"
   # model: "FastSAM-s"
-  force_collect: False
+  force_collect: False
diff --git a/configs/training/config.yaml b/configs/training/config.yaml
@@ -8,10 +8,11 @@ lr: 0.001
 batch_size: 32
 time_steps: 10
 num_objects: 8
-name: debug-training
+name: final-training
 game: Pong
-model: SAM
+model: FastSAM-x
 num_iterations: 1000
 ground_truth_masks: false
+t: 0
 
 save_models: false
diff --git a/configs/training/predictor/baseline.yaml b/configs/training/predictor/baseline.yaml
@@ -1 +1,2 @@
-_target_: src.model.predictor_baseline.PredictorBaseline
+_target_: src.model.predictor_baseline.PredictorBaseline
+num_actions: 18
diff --git a/configs/training/predictor/residual.yaml b/configs/training/predictor/residual.yaml
@@ -1,2 +1,3 @@
 _target_: src.model.residual_predictor.ResidualPredictor
-num_layers: 1
+num_layers: 1
+num_actions: 18