1 Star 0 Fork 0

VRxiaojie/AIfence

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
1.py 402 Bytes
一键复制 编辑 原始数据 按行查看 历史
VRxiaojie 提交于 2023-04-10 09:29 . first commit
import torch
# Load the model
model = torch.load('runs/train/exp11/weights/best.pt')
# Print the type of each object in the model
for obj in model:
print(obj)
print(type(obj))
# Remove unsupported Lua types
# For example, if the model contains a Lua string, you can remove it with:
del model['string_object']
# Save the modified model
torch.save(model, 'runs/train/exp11/weights/fixed.pt')
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/vrxiaojie/aifence.git
[email protected]:vrxiaojie/aifence.git
vrxiaojie
aifence
AIfence
master

搜索帮助