ImportError: cannot import name 'Caffe2Tracer' from 'detectron2.export'

[paul-ml75]

Hey guys!

i'm using this command on export_model.py in order to save a torchscript model:

python3 tools/deploy/export_model.py --config-file configs/Base-RetinaNet.yaml --format torchscript --output test

But I get this error:

File "/transfiner-main/tools/deploy/export_model.py", line 14, in
from detectron2.export import (
ImportError: cannot import name 'Caffe2Tracer' from 'detectron2.export' (/Library/Frameworks/Python.framework/Versions/3.10/lib/python3.10/site-packages/detectron2/export/init.py)

I'm using pytorch v2.

do you have any idea how to fix this issue I got?

thank you,

Paul

[lkeab]

Hi, this project is supported in pytorch==1.7.1. I am not sure for the pytorch v2 support. You can refer to the discussion here for detectron2 support.