Skip to content

Y
yolov5
提交 546f9735 authored 作者: Glenn Jocher's avatar Glenn Jocher
浏览文件
操作

model fuse

上级 8fe299f1
隐藏空白字符变更
内嵌 并排
正在显示 包含 0 行增加 和 2 行删除
test.py
浏览文件 @ 546f9735
...@@ -33,8 +33,6 @@ def test(data, ...@@ -33,8 +33,6 @@ def test(data,
google_utils.attempt_download(weights) google_utils.attempt_download(weights)
model = torch.load(weights, map_location=device)['model'] model = torch.load(weights, map_location=device)['model']
torch_utils.model_info(model) torch_utils.model_info(model)
# Fuse
# model.fuse() # model.fuse()
model.to(device) model.to(device)
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论