Spaces:
Running
Running
| { | |
| "task": "006_train_vrt_videodeblurring_gopro" | |
| , "model": "vrt" | |
| , "gpu_ids": [0,1,2,3,4,5,6,7] | |
| , "dist": true | |
| , "find_unused_parameters": false | |
| , "use_static_graph": true | |
| ,"scale": 1 | |
| , "n_channels": 3 | |
| , "path": { | |
| "root": "experiments" | |
| , "pretrained_netG": null | |
| , "pretrained_netE": null | |
| } | |
| , "datasets": { | |
| "train": { | |
| "name": "train_dataset" | |
| , "dataset_type": "VideoRecurrentTrainDataset" | |
| , "dataroot_gt": "trainsets/GoPro/train_GT.lmdb" | |
| , "dataroot_lq": "trainsets/GoPro/train_GT_blurred.lmdb" | |
| , "meta_info_file": "data/meta_info/meta_info_GoPro_train_GT.txt" | |
| , "filename_tmpl": "06d" | |
| , "filename_ext": "png" | |
| , "test_mode": false | |
| , "io_backend": {"type": "lmdb"} | |
| , "num_frame": 6 | |
| , "gt_size": 192 | |
| , "interval_list": [1] | |
| , "random_reverse": false | |
| , "use_hflip": true | |
| , "use_rot": true | |
| , "dataloader_shuffle": true | |
| , "dataloader_num_workers": 32 | |
| , "dataloader_batch_size": 8 | |
| } | |
| , "test": { | |
| "name": "test_dataset" | |
| , "dataset_type": "VideoRecurrentTestDataset" | |
| , "dataroot_gt": "testsets/GoPro11/test_GT" | |
| , "dataroot_lq": "testsets/GoPro11/test_GT_blurred" | |
| , "cache_data": false | |
| , "io_backend": {"type": "disk"} | |
| , "num_frame": -1 | |
| } | |
| } | |
| , "netG": { | |
| "net_type": "vrt" | |
| , "upscale": 1 | |
| , "img_size": [6,192,192] | |
| , "window_size": [6,8,8] | |
| , "depths": [8,8,8,8,8,8,8, 4,4, 4,4] | |
| , "indep_reconsts": [9,10] | |
| , "embed_dims": [96,96,96,96,96,96,96, 120,120, 120,120] | |
| , "num_heads": [6,6,6,6,6,6,6, 6,6, 6,6] | |
| , "spynet_path": "model_zoo/vrt/spynet_sintel_final-3d2a1287.pth" // automatical download | |
| , "pa_frames": 2 | |
| , "deformable_groups": 16 | |
| , "nonblind_denoising": false | |
| , "use_checkpoint_attn": true | |
| , "use_checkpoint_ffn": true | |
| , "no_checkpoint_attn_blocks": [2,3,4] | |
| , "no_checkpoint_ffn_blocks": [1,2,3,4,5,9] | |
| , "init_type": "default" | |
| } | |
| , "train": { | |
| "G_lossfn_type": "charbonnier" | |
| , "G_lossfn_weight": 1.0 | |
| , "G_charbonnier_eps": 1e-9 | |
| , "E_decay": 0 // Exponential Moving Average for netG: set 0 to disable; default setting 0.999 | |
| , "G_optimizer_type": "adam" // fixed, adam is enough | |
| , "G_optimizer_lr": 4e-4 // learning rate | |
| , "G_optimizer_betas": [0.9,0.99] | |
| , "G_optimizer_wd": 0 // weight decay, default 0 | |
| , "G_optimizer_clipgrad": null // unused | |
| , "G_optimizer_reuse": true // | |
| , "fix_iter": 20000 | |
| , "fix_lr_mul": 0.125 | |
| , "fix_keys": ["spynet", "deform"] | |
| , "total_iter": 300000 | |
| , "G_scheduler_type": "CosineAnnealingWarmRestarts" | |
| , "G_scheduler_periods": 300000 | |
| , "G_scheduler_eta_min": 1e-7 | |
| , "G_regularizer_orthstep": null // unused | |
| , "G_regularizer_clipstep": null // unused | |
| , "G_param_strict": true | |
| , "E_param_strict": true | |
| , "checkpoint_test": 5000 // for testing | |
| , "checkpoint_save": 5000 // for saving model | |
| , "checkpoint_print": 200 // for print | |
| } | |
| , "val": { | |
| "save_img": false | |
| , "pad_seq": false | |
| , "flip_seq": false | |
| , "center_frame_only": false | |
| , "num_frame_testing": 18 | |
| , "num_frame_overlapping": 2 | |
| , "size_patch_testing": 192 | |
| } | |
| } | |