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: