Fixed net.py

2021-06-28 01:21:30 +02:00 · 2021-06-28 01:21:30 +02:00 · 5870848da9
commit 5870848da9
parent aa2a248503
1 changed files with 3 additions and 2 deletions
--- a/monoloco/network/net.py
+++ b/monoloco/network/net.py
@ -100,6 +100,7 @@ class Loco:
            self.model = model
        self.model.eval()  # Default is train
        self.model.to(self.device)
+        if casr:
            self.turning_model.eval()  # Default is train
            self.turning_model.to(self.device)