search

bubbliiiing / deeplabv3-plus-pytorch

这是一个deeplabv3-plus-pytorch的源码，可以用于训练自己的模型。
MIT License
920 stars 164 forks source link

model convert to TorchScript #86

Open jinyu-118 opened 1 year ago

jinyu-118 commented 1 year ago

I would like to ask if it is possible to provide a GPU TorchScript model conversion instead of the onnx model

jinyu-118 commented 1 year ago

尝试了自己转换模型，但是失败了。