torchscripted model for training phase

zhiqwang / yolort

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.

https://zhiqwang.com/yolort

GNU General Public License v3.0

716 stars 153 forks source link

torchscripted model for training phase #131

Open fermat97 opened 3 years ago

fermat97 commented 3 years ago

Hi. Is it possible to use TorchScripted yolort model for the training phase instead of inference?

zhiqwang commented 3 years ago

Hi @fermat97 , yolort are designed for supporting training with torchscript, but now there are some bugs in the trainer. You can check #60 for the development progress.