modify

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
3 years ago · 35a1502c6d
1 changed files with 2 additions and 0 deletions
--- a/tsm.py
+++ b/tsm.py
@ -95,6 +95,8 @@ class Tsm(NNOperator):
                    )
        inputs = data.to(self.device)[None, ...]

+        self.model.eval()
+
        feats = self.model.forward_features(inputs)
        if self.model.reshape:
            if self.model.is_shift and self.model.temporal_pool: