From 5870848da903bb69585ba48e6b6f7fe83ba1976e Mon Sep 17 00:00:00 2001
From: Charles Joseph Pierre Beauville <charlesbeauville@gmail.com>
Date: Mon, 28 Jun 2021 01:21:30 +0200
Subject: [PATCH] Fixed net.py

---
 monoloco/network/net.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/monoloco/network/net.py b/monoloco/network/net.py
index b27e3a3..217384c 100644
--- a/monoloco/network/net.py
+++ b/monoloco/network/net.py
@@ -100,8 +100,9 @@ class Loco:
             self.model = model
         self.model.eval()  # Default is train
         self.model.to(self.device)
-        self.turning_model.eval()  # Default is train
-        self.turning_model.to(self.device)
+        if casr:
+            self.turning_model.eval()  # Default is train
+            self.turning_model.to(self.device)
 
     def forward(self, keypoints, kk, keypoints_r=None):
         """