From 6e447bdc8c3705a633178c9023737a7195f8ac3b Mon Sep 17 00:00:00 2001 From: gexy5 Date: Mon, 13 Jun 2022 18:58:18 +0800 Subject: [PATCH] modify README Signed-off-by: gexy5 --- tsm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tsm.py b/tsm.py index 8e00784..e7eeefe 100644 --- 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,