Kwangyeon Gill

Convert a model with custom pytorch CUDA kernel

1

# Ask a Question ### Question I have implemented a custom pytorch CUDA kernel (and CPU kernel) for rotated bounding boxes that is not supported in pytorch. (like this example...

references/segmentation/coco_utils might require merging rles?

1

https://github.com/pytorch/vision/blob/6d7851bd5e2bedc294e40e90532f0e375fcfee04/references/segmentation/coco_utils.py#L27-L41 Above seems to assume that objects are not occluded, not merging rles from `frPyObjects`. In such case, i think it must be changed to ```python rles = coco_mask.frPyObjects(polygons, height,...

Weird bug when setting `val_check_interval` dynamically in `setup()`

4

### Bug description I want to dynamically set `val_check_interval` based on the total number of training steps. Specifically, i calculate val_check_interval using self.trainer.estimated_stepping_batches // 10 in the setup() method, aiming...

Torchvision Faster R-CNN onnx export with dynamic batch size fails during inference

2

### 🐛 Describe the bug ```python from torchvision import models frcnn = models.detection.fasterrcnn_resnet50_fpn_v2(pretrained=True) import io import torch x = torch.rand(4, 3, 224, 224) with io.BytesIO() as f: torch.onnx.export( frcnn, x,...