logo
Browse Source

modify

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5 2 years ago
parent
commit
35a1502c6d
  1. 2
      tsm.py

2
tsm.py

@ -95,6 +95,8 @@ class Tsm(NNOperator):
) )
inputs = data.to(self.device)[None, ...] inputs = data.to(self.device)[None, ...]
self.model.eval()
feats = self.model.forward_features(inputs) feats = self.model.forward_features(inputs)
if self.model.reshape: if self.model.reshape:
if self.model.is_shift and self.model.temporal_pool: if self.model.is_shift and self.model.temporal_pool:

Loading…
Cancel
Save