model_params: common_params: num_kp: 15 image_channel: 3 feature_channel: 32 estimate_jacobian: False # True kp_detector_params: temperature: 0.1 block_expansion: 32 max_features: 1024 scale_factor: 0.25 # 0.25 num_blocks: 5 reshape_channel: 16384 # 16384 = 1024 * 16 reshape_depth: 16 he_estimator_params: block_expansion: 64 max_features: 2048 num_bins: 66 generator_params: block_expansion: 64 max_features: 512 num_down_blocks: 2 reshape_channel: 32 reshape_depth: 16 # 512 = 32 * 16 num_resblocks: 6 estimate_occlusion_map: True dense_motion_params: block_expansion: 32 max_features: 1024 num_blocks: 5 reshape_depth: 16 compress: 4 discriminator_params: scales: [1] block_expansion: 32 max_features: 512 num_blocks: 4 sn: True mapping_params: coeff_nc: 70 descriptor_nc: 1024 layer: 3 num_kp: 15 num_bins: 66