File size: 726 Bytes
16805ce | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | attention_mode: hier_hub
camera_dim: 12
camera_dropout_p: 0.1
group_label_player_ids: null
hier_direct_frac: 1.0
hier_direct_layers: null
hier_direct_schedule: uniform
hier_group_format: membership
hier_group_mode: csv
hier_group_seed: 0
hier_random_preserve_sizes: 1
local_attn_size: 7
local_attn_size_after: 21
local_attn_switch_step: 10000
loss_type: diffusion_forcing
num_frame_per_block: 3
num_global_hub_tokens: 6
num_groups: 2
num_hub_tokens: 16
num_local_hub_tokens: 5
num_players: 4
simplex_gamma: 1.0
simplex_pool_size: 10
target_rgb_frames: 165
train_step_offset: 10000
training_stage: stage2
use_camera_pose: true
use_simplex_rope: true
vae_latent_channels: 48
hier_local_hub_mode: shared
hier_no_local_hub: false
|