| { |
| "data": { |
| "at_cfg": { |
| "cj": true, |
| "cj_bri": 0.3, |
| "cj_con": 0.3, |
| "cj_hue": 0.1, |
| "cj_sat": 0.3, |
| "gamma": false, |
| "gblur": true |
| }, |
| "data_aug": { |
| "crop": true, |
| "hflip": true, |
| "para_crop": [384, 832], |
| "swap": true |
| }, |
| "epoches_raw": 100, |
| "epoches_ft": -1, |
| "root_sintel": "YOUR_DIR/Sintel/", |
| "root_sintel_raw": "YOUR_DIR/Sintel-raw/", |
| "full_seg_root_sintel": "YOUR_DIR/full_seg/Sintel", |
| "full_seg_root_sintel_raw": "YOUR_DIR/full_seg/Sintel-raw", |
| "key_obj_root_sintel": "YOUR_DIR/key_objects/Sintel", |
| "key_obj_root_sintel_raw": "YOUR_DIR/key_objects/Sintel-raw", |
| "run_at": true, |
| "test_shape": [448, 1024], |
| "train_subsplit": "trainval", |
| "type": "Sintel_Raw+ft_2stage", |
| "val_subsplit": "trainval" |
| }, |
| "loss": { |
| "edge_aware_alpha": 10, |
| "occ_from_back": true, |
| "smooth_type": "2nd", |
| "smooth_edge": "image", |
| "type": "unflow", |
| "w_l1": 0.15, |
| "w_ph_scales": [1.0, 1.0, 1.0, 1.0, 0.0], |
| "w_sm": 0, |
| "w_ssim": 0.85, |
| "w_ternary": 0.0, |
| "warp_pad": "border", |
| "with_bk": true |
| }, |
| "model": { |
| "learned_upsampler": true, |
| "reduce_dense": true, |
| "type": "pwclite" |
| }, |
| "seed": 42, |
| "train": { |
| "ar_eps": 0.0, |
| "ar_q": 1.0, |
| "batch_size": 8, |
| "beta": 0.999, |
| "bias_decay": 0, |
| "epoch_num": 200, |
| "epoch_size": 1000, |
| "key_obj_aug": false, |
| "lr": 0.0002, |
| "lr_scheduler": { |
| "module": "OneCycleLR", |
| "params": { |
| "max_lr": 0.0004, |
| "pct_start": 0.05, |
| "cycle_momentum": false, |
| "anneal_strategy": "linear" |
| } |
| }, |
| "mask_st": true, |
| "max_grad_norm": 10, |
| "momentum": 0.9, |
| "n_gpu": 8, |
| "optim": "adam", |
| "pretrained_model": null, |
| "print_freq": 100, |
| "record_freq": 500, |
| "run_atst": false, |
| "run_ot": false, |
| "run_st": false, |
| "save_iter": 10000, |
| "st_cfg": { |
| "add_noise": true, |
| "hflip": true, |
| "rotate": [-0.2, 0.2, -0.015, 0.015], |
| "squeeze": [0.86, 1.16, 1.0, 1.0], |
| "trans": [0.2, 0.015], |
| "vflip": true, |
| "zoom": [1.0, 1.5, 0.985, 1.015] |
| }, |
| "stage1": { |
| "epoch": 50, |
| "loss": { |
| "occ_from_back": false, |
| "w_l1": 0.0, |
| "w_ssim": 0.0, |
| "w_ternary": 1.0 |
| }, |
| "train": { |
| "ot_size": [320, 704], |
| "run_atst": true, |
| "run_ot": true, |
| "run_st": true |
| } |
| }, |
| "val_epoch_size": 5, |
| "valid_size": 0, |
| "w_ar": 0.02, |
| "weight_decay": 1e-06, |
| "workers": 8 |
| }, |
| "trainer": "SINTEL_AR" |
| } |
|
|
