sintel_ft_ar_active.json

{"data": {"at_cfg": {"cj": true,
                     "cj_bri": 0.4,
                     "cj_con": 0.4,
                     "cj_hue": 0.16,
                     "cj_sat": 0.4,
                     "gamma": true,
                     "gblur": true},
          "label_ratio": 0.1,
          "root_sintel": "/PATH/TO/MPI-Sintel",
          "run_at": true,
          "sample_scores_sorted": "/PATH/TO/YOUR/model_ep500_ckpt_sintel_train_occ_ratio.txt",
          "test_shape": [448, 1024],
          "train_n_frames": 2,
          "train_subsplit": "train",
          "type": "Sintel",
          "val_n_frames": 2,
          "val_subsplit": "val"},
 "data_aug": {"crop": true,
              "hflip": true,
              "para_crop": [384, 768],
              "para_rescale": {"min_scale": -0.2,
                               "max_scale": 0.6 },
              "rescale": true,
              "swap": false},
 "loss": {"charbonnier_epsilon": 0.01,
          "charbonnier_q": 0.4,
          "edge_aware_alpha": 10,
          "occ_from_back": false,
          "sup_type": "L1",
          "type": "semiflow",
          "w_alpha": 1,
          "w_l1": 0.0,
          "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.0,
          "w_sup_scales": [0.32, 0.08, 0.02, 0.01, 0.005],
          "w_ternary": 1.0,
          "warp_pad": "border",
          "with_bk": true},
 "model": {"n_frames": 2,
           "reduce_dense": true,
           "type": "pwclite",
           "upsample": true},
 "seed": 0,
 "train": {"ar_eps": 0.0,
           "ar_q": 1.0,
           "batch_size": 8,
           "beta": 0.999,
           "bias_decay": 0,
           "epoch_num": 1000,
           "epoch_size": 1000,
           "lr": 0.0001,
           "lr_scheduler": {"module": "MultiStepLR",
                            "params": {"milestones": [400, 600, 800], 
                                       "gamma": 0.5}},
           "mask_st": true,
           "max_grad_norm": 5,
           "momentum": 0.9,
           "n_gpu": 2,
           "optim": "adam",
           "ot_compact": 50,
           "ot_fast": true,
           "ot_size": [320, 704],
           "ot_slic": true,
           "pretrained_model": "/PATH/TO/YOUR/PRETRAINED/model_ep500_ckpt.pth.tar",
           "print_freq": 100,
           "record_freq": 500,
           "run_atst": true,
           "run_ot": false,
           "run_st": true,
           "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]},
           "val_epoch_size": 5,
           "valid_size": 0,
           "w_ar": 0.01,
           "weight_decay": 1e-06,
           "workers": 8},
 "trainer": "Sintel_AR"}