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

Update train.py

Browse Source
This commit is contained in:
hzwer
2022-11-17 11:31:43 +08:00
committed by GitHub
parent d430234d81
commit 58f23dc85a
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@@ -40,6 +40,9 @@ def train(model, local_rank):
if local_rank == 0: if local_rank == 0:
writer = SummaryWriter('train') writer = SummaryWriter('train')
writer_val = SummaryWriter('validate') writer_val = SummaryWriter('validate')
else:
writer = None
writer_val = None
step = 0 step = 0
nr_eval = 0 nr_eval = 0
dataset = VimeoDataset('train') dataset = VimeoDataset('train')