modify README

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
2 years ago · 6e447bdc8c
1 changed files with 1 additions and 1 deletions
--- a/tsm.py
+++ b/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.eval()
        self.transform_cfgs = get_configs(
-                                    side_size=224,
+                                    side_size=256,
                                    crop_size=224,
                                    num_frames=8,
                                    mean=self.model.input_mean,