inline c10::intrusive_ptr<ivalue::Tuple> IValue::toTuple() const & {
AT_ASSERT(isTuple(), "Expected Tuple but got ", tagKind());
return toIntrusivePtr<ivalue::Tuple>();
}
这里先说说我是怎么导出.pt模型的,yolov5官方提供了导出的python的代码,在如下:
问题就出现这个转换的代码里,作者说,这个转换默认是调用cpu 的,所以如何使用GPU需要重新修改一下:
你需要修改几句:
github作者的回复:https://github.com/yasenh/libtorch-yolov5#torchscript-model-export