olegsij

Converted to TorchScript model outputs mask with wrong shape

2

**Describe the bug** Converted to torchscript model outputs masks of size (128, 128) instead of (512, 512) **Reproduction** 1. What command or script did you run? ``` python tools/deployment/pytorch2torchscript.py PATH_TO_CONFIG...