| dataset_params: | |
| root_dir: ../moving-gif | |
| frame_shape: null | |
| id_sampling: False | |
| augmentation_params: | |
| flip_param: | |
| horizontal_flip: True | |
| time_flip: True | |
| crop_param: | |
| size: [256, 256] | |
| resize_param: | |
| ratio: [0.9, 1.1] | |
| jitter_param: | |
| hue: 0.5 | |
| model_params: | |
| common_params: | |
| num_tps: 10 | |
| num_channels: 3 | |
| bg: False | |
| multi_mask: True | |
| generator_params: | |
| block_expansion: 64 | |
| max_features: 512 | |
| num_down_blocks: 3 | |
| dense_motion_params: | |
| block_expansion: 64 | |
| max_features: 1024 | |
| num_blocks: 5 | |
| scale_factor: 0.25 | |
| avd_network_params: | |
| id_bottle_size: 128 | |
| pose_bottle_size: 128 | |
| train_params: | |
| num_epochs: 100 | |
| num_repeats: 50 | |
| epoch_milestones: [70, 90] | |
| lr_generator: 2.0e-4 | |
| batch_size: 28 | |
| scales: [1, 0.5, 0.25, 0.125] | |
| dataloader_workers: 12 | |
| checkpoint_freq: 50 | |
| dropout_epoch: 35 | |
| dropout_maxp: 0.5 | |
| dropout_startp: 0.2 | |
| dropout_inc_epoch: 10 | |
| bg_start: 0 | |
| transform_params: | |
| sigma_affine: 0.05 | |
| sigma_tps: 0.005 | |
| points_tps: 5 | |
| loss_weights: | |
| perceptual: [10, 10, 10, 10, 10] | |
| equivariance_value: 10 | |
| warp_loss: 10 | |
| bg: 10 | |
| train_avd_params: | |
| num_epochs: 100 | |
| num_repeats: 50 | |
| batch_size: 256 | |
| dataloader_workers: 24 | |
| checkpoint_freq: 10 | |
| epoch_milestones: [70, 90] | |
| lr: 1.0e-3 | |
| lambda_shift: 1 | |
| lambda_affine: 1 | |
| random_scale: 0.25 | |
| visualizer_params: | |
| kp_size: 5 | |
| draw_border: True | |
| colormap: 'gist_rainbow' |