浏览代码

model fuse

5.0
Glenn Jocher 4 年前
父节点
当前提交
546f973562
共有 1 个文件被更改,包括 0 次插入2 次删除
  1. +0
    -2
      test.py

+ 0
- 2
test.py 查看文件

@@ -33,8 +33,6 @@ def test(data,
google_utils.attempt_download(weights)
model = torch.load(weights, map_location=device)['model']
torch_utils.model_info(model)

# Fuse
# model.fuse()
model.to(device)


正在加载...
取消
保存