Browse Source
modify README
Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5
2 years ago
1 changed files with
1 additions and
1 deletions
-
tsm.py
|
|
@ -62,7 +62,7 @@ class Tsm(NNOperator): |
|
|
self.model = create_model(model_name=model_name, pretrained=True, weights_path=self.weights_path, device=self.device) |
|
|
self.model = create_model(model_name=model_name, pretrained=True, weights_path=self.weights_path, device=self.device) |
|
|
self.model.eval() |
|
|
self.model.eval() |
|
|
self.transform_cfgs = get_configs( |
|
|
self.transform_cfgs = get_configs( |
|
|
side_size=224, |
|
|
|
|
|
|
|
|
side_size=256, |
|
|
crop_size=224, |
|
|
crop_size=224, |
|
|
num_frames=8, |
|
|
num_frames=8, |
|
|
mean=self.model.input_mean, |
|
|
mean=self.model.input_mean, |
|
|
|
