Closed
Description
Feature context.
This feature tracks the progress in implementing the fx2trt converters using the torch2trt converters
Task steps:
-
Run the unit tests in TensorRT TensorRT/py/torch_tensorrt/fx/test/aten_op/*.py to study acc and aten trace and how they differ
STATUS : done -
Run progressively the model list and start supporting all the ops in the models. The model list:
squeezenet1_0
mobilenet_v2
inception_v3
efficientnet_b0
regnet_y_8gf
STATUS: At present the test exists for resnet18 /TensorRT/py/torch_tensorrt/fx/test/tracer/test_resnet.py. Expand this for the above networks -
In the process, integrate the converters such that they are agnostic to the aten and acc tracers. So we would want to make the flow generic to fx2trt
STATUS : In progress