Currently, this will fail with RuntimeError: Conversion of function torchvision.ops.roi_align.roi_align not supported.. Do one of the following:
a. Create a acc_op wrapper for torchvision.ops.roi_align.roi_align and implement a converter for it in acc_ops_converters.
b. Dont bother with acc wrapper, just create a torchvision_converters file and directly implement a converter for torchvision.ops.roi_align.roi_align
Here is some sample test code to help with the acc_op implementation
Currently, this will fail with
RuntimeError: Conversion of function torchvision.ops.roi_align.roi_align not supported.
. Do one of the following: a. Create a acc_op wrapper fortorchvision.ops.roi_align.roi_align
and implement a converter for it in acc_ops_converters. b. Dont bother with acc wrapper, just create atorchvision_converters
file and directly implement a converter fortorchvision.ops.roi_align.roi_align