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()