CASR model passed through args with std gestures

2021-06-28 10:04:31 +02:00 · 2021-06-28 10:04:31 +02:00 · 8ac304a430
commit 8ac304a430
parent 872aec93e1
1 changed files with 4 additions and 1 deletions
--- a/monoloco/network/net.py
+++ b/monoloco/network/net.py
@ -61,6 +61,9 @@ class Loco:
        if casr == 'std':
            print("CASR with standard gestures")
            turning_output_size = 3
+            if casr_model:
+                turning_model_path = casr_model
+            else:
                turning_model_path = "/home/beauvill/Repos/monoloco/data/outputs/casr_standard-210613-0005.pkl"
        elif casr == 'nonstd':
            turning_output_size = 4