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,