From 35a1502c6d4cd8322e08b30739370c535e6e9555 Mon Sep 17 00:00:00 2001
From: gexy5 <xinyu.ge@zilliz.com>
Date: Mon, 13 Jun 2022 17:22:42 +0800
Subject: [PATCH] modify

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
---
 tsm.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tsm.py b/tsm.py
index edab8f4..4bcdc87 100644
--- 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: