Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2025-12-15 16:07:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update HD_multi_4X.py

Browse Source
This commit is contained in:
hzwer
2021-11-15 23:41:25 +08:00
committed by GitHub
parent cecf30a0aa
commit d66318d938
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/HD_multi_4X.py
View File

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