mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-17 00:23:14 +00:00
Justin John
46 lines
1.1 KiB
YAML
46 lines
1.1 KiB
YAML
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: 73
|
|
descriptor_nc: 1024
|
|
layer: 3
|
|
num_kp: 15
|
|
num_bins: 66
|
|
|