tangjunjun

摘要： 若使用tensorrt加载wts格式，需将模型训练的pt、pth、ckpt等格式权重转换为wts，其代码细节如下： def checkpint2wts(model, wts_file): ''' model:模型,需要权重 wts_file：保存wts权重路径，如result.wts ''' imp 阅读全文