ソースを参照

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)


読み込み中…
キャンセル
保存