-
Notifications
You must be signed in to change notification settings - Fork 3
/
sintel_raw.json
executable file
·53 lines (53 loc) · 1.68 KB
/
sintel_raw.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
{"data": {"root_sintel": "/PATH/TO/MPI-Sintel",
"root_sintel_raw": "/PATH/TO/sintel_raw",
"run_at": false,
"test_shape": [448, 1024],
"train_n_frames": 2,
"type": "Sintel_Raw",
"val_n_frames": 2,
"val_subsplit": "val"},
"data_aug": {"crop": true,
"hflip": true,
"para_crop": [384, 832],
"resize": false,
"swap": true},
"loss": {"edge_aware_alpha": 10,
"occ_from_back": true,
"type": "unflow",
"w_l1": 0.15,
"w_ph_scales": [1.0, 1.0, 1.0, 1.0, 0.0],
"w_sm_scales": [1.0, 0.0, 0.0, 0.0, 0.0],
"w_smooth": 50.0,
"w_ssim": 0.85,
"w_ternary": 0.0,
"warp_pad": "border",
"with_bk": true},
"model": {"n_frames": 2,
"reduce_dense": true,
"type": "pwclite",
"upsample": true},
"seed": 0,
"train": {"batch_size": 8,
"beta": 0.999,
"bias_decay": 0,
"epoch_num": 500,
"epoch_size": 1000,
"lr": 0.0001,
"max_grad_norm": 100,
"momentum": 0.9,
"n_gpu": 2,
"optim": "adam",
"pretrained_model": null,
"print_freq": 100,
"record_freq": 500,
"save_iter": 10000,
"stage1": {"epoch": 50,
"loss": {"occ_from_back": false,
"w_l1": 0.0,
"w_ssim": 0.0,
"w_ternary": 1.0}},
"val_epoch_size": 5,
"valid_size": 0,
"weight_decay": 1e-06,
"workers": 8},
"trainer": "Sintel"}