Browse Source
modify
Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5
3 years ago
1 changed files with
2 additions and
0 deletions
-
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: |
|
|