diff options
| author | StevenLiuWen <liuwen@shanghaitech.edu.cn> | 2018-03-13 03:28:06 -0400 |
|---|---|---|
| committer | StevenLiuWen <liuwen@shanghaitech.edu.cn> | 2018-03-13 03:28:06 -0400 |
| commit | fede6ca1dd0077ff509d84bd24028cc7a93bb119 (patch) | |
| tree | af7f6e759b5dec4fc2964daed09e903958b919ed /Codes/flownet2/src/training_schedules.py | |
first commit
Diffstat (limited to 'Codes/flownet2/src/training_schedules.py')
| -rw-r--r-- | Codes/flownet2/src/training_schedules.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/Codes/flownet2/src/training_schedules.py b/Codes/flownet2/src/training_schedules.py new file mode 100644 index 0000000..4db5aab --- /dev/null +++ b/Codes/flownet2/src/training_schedules.py @@ -0,0 +1,12 @@ +LONG_SCHEDULE = { + 'step_values': [400000, 600000, 800000, 1000000], + 'learning_rates': [0.0001, 0.00005, 0.000025, 0.0000125, 0.00000625], + 'momentum': 0.9, + 'momentum2': 0.999, + 'weight_decay': 0.0004, + 'max_iter': 1200000, +} + +FINETUNE_SCHEDULE = { + # TODO: Finetune schedule +} |