mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-02-04 04:34:41 +08:00
33 lines
721 B
JSON
33 lines
721 B
JSON
{
|
|
"seed": 2020,
|
|
"save_dir": "release_model/",
|
|
"data_loader": {
|
|
"name": "youtube-vos",
|
|
"data_root": "datasets_sttn/",
|
|
"w": 640,
|
|
"h": 120,
|
|
"sample_length": 5
|
|
},
|
|
"losses": {
|
|
"hole_weight": 1,
|
|
"valid_weight": 1,
|
|
"adversarial_weight": 0.01,
|
|
"GAN_LOSS": "hinge"
|
|
},
|
|
"trainer": {
|
|
"type": "Adam",
|
|
"beta1": 0,
|
|
"beta2": 0.99,
|
|
"lr": 1e-4,
|
|
"d2glr": 1,
|
|
"batch_size": 8,
|
|
"num_workers": 2,
|
|
"verbosity": 2,
|
|
"log_step": 100,
|
|
"save_freq": 1e4,
|
|
"valid_freq": 1e4,
|
|
"iterations": 50e4,
|
|
"niter": 15e4,
|
|
"niter_steady": 30e4
|
|
}
|
|
} |