Centernet2 to torch.jit.trace

xingyizhou / CenterNet2

Two-stage CenterNet

Apache License 2.0

1.19k stars 189 forks source link

Open djaym7 opened 3 years ago

djaym7 commented 3 years ago

It is throwing whole bunch of errors where constants are used in the custom heads/backbones. Can we get a jit.trace- able centernet2 ?

xingyizhou commented 3 years ago

Thank you for the question/ suggestion. However I don't have a background in this ...

djaym7 commented 3 years ago