From d66318d938e3397dd389a88146873f37726ee739 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Mon, 15 Nov 2021 23:41:25 +0800
Subject: [PATCH] Update HD_multi_4X.py

---
 benchmark/HD_multi_4X.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmark/HD_multi_4X.py b/benchmark/HD_multi_4X.py
index 284b091..a089002 100644
--- a/benchmark/HD_multi_4X.py
+++ b/benchmark/HD_multi_4X.py
@@ -14,7 +14,7 @@ from yuv_frame_io import YUV_Read,YUV_Write
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model = Model(arbitrary=True)
-model.load_model('train_log')
+model.load_model('RIFE_m_train_log')
 model.eval()
 model.device()