Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-24 04:19:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
hzwer
2022-03-15 15:37:06 +08:00
committed by GitHub
parent 7788494700
commit f114857833
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@@ -21,10 +21,10 @@ log_path = 'train_log'
def get_learning_rate(step):
if step < 2000:
mul = step / 2000.
return 1e-4 * mul
return 3e-4 * mul
else:
mul = np.cos((step - 2000) / (args.epoch * args.step_per_epoch - 2000.) * math.pi) * 0.5 + 0.5
return (1e-4 - 1e-5) * mul + 1e-5
return (3e-4 - 3e-5) * mul + 3e-5
def flow2rgb(flow_map_np):
h, w, _ = flow_map_np.shape