Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,224 Bytes
352b049 |
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 |
if_resume_training: false
debug: false
device: cuda
load: false
load_path: ''
dataset_main_path: ./datasets
dataset_annotation_path: ./datasets/annotations&interm
dataset_precomputed_path: ./datasets/processed
_load_saved_aoutoenc: hmp-amass
output_log_path: ./models/
slurm_id: 0
slurm_first_run: None
info: cvpr_release
pretrained_autoencoder_path: ./models/checkpoint_300.pt
lr: 0.001
diffusion_objective: pred_x0
weight_decay: 0.0
if_use_ema: true
step_start_ema: 100
ema_power: 0.6666666666666666
ema_update_every: 10
ema_min_value: 0.0
use_lr_scheduler: true
lr_scheduler_kwargs:
lr_scheduler_type: ExponentialLRSchedulerWarmup
warmup_duration: 75
update_every: 2
min_lr: 0.0001
gamma_decay: 0.98
decay_factor: 0.1
patience: 1
diffusion_conditioning: true
num_epochs: 150
num_workers: 4
batch_size: 64
batch_size_eval: 256
eval_frequency: 1
train_pick_best_sample_among_k: 50
diffusion_k_train_after_epochs: 0
diffusion_k_train_monot_increase: false
diffusion_k_train_monot_increase_howmanyperepoch: 1
similarity_space: input_space
diffusion_activation: identity
num_prob_samples: 50
diffusion_timesteps: 10
diffusion_type: NonisotropicGaussianDiffusion
diffusion_loss_type: snr
loss_reduction_type: l1
if_run_as_isotropic: false
if_sigma_n_scale: true
diffusion_covariance_type: skeleton-diffusion
gamma_scheduler: cosine
beta_schedule: cosine
sigma_n_scale: spectral
num_iter_perepoch: null
seed: 63485
motion_repr_type: SkeletonRescalePose
dataset_type: AMASSDataset
dataset_name: amass
latent_size: 96
output_size: 3
z_activation: tanh
obs_length: 30
prediction_horizon_train: 120
prediction_horizon_eval: 120
prediction_horizon: 120
pred_length: 120
prediction_horizon_train_min: 10
prediction_horizon_train_min_from_epoch: 200
curriculum_it: 10
random_prediction_horizon: true
if_consider_hip: false
history_sec: 0.5
prediction_horizon_sec: 2
pose_box_size: 1.2
seq_centering: 0
task_name: hmp
num_joints: 22
fps: 60
multimodal_threshold: 0.4
precomputed_folder: ./models/
annotations_folder: ./models/
data_loader_train:
stride: 60
augmentation: 30
shuffle: true
datasets:
- ACCAD
- BMLhandball
- BMLmovi
- BMLrub
- EKUT
- CMU
- EyesJapanDataset
- KIT
- PosePrior
- TCDHands
- TotalCapture
da_mirroring: 0.5
da_rotations: 1.0
drop_last: true
if_load_mmgt: false
data_loader_train_eval:
stride: 30
augmentation: 0
shuffle: false
datasets:
- ACCAD
- BMLhandball
- BMLmovi
- BMLrub
- EKUT
- CMU
- EyesJapanDataset
- KIT
- PosePrior
- TCDHands
- TotalCapture
da_mirroring: 0.0
da_rotations: 0.0
drop_last: false
if_load_mmgt: false
data_loader_valid:
stride: 30
augmentation: 0
shuffle: false
datasets:
- HumanEva
- HDM05
- SFU
- MoSh
file_idces: all
drop_last: false
if_load_mmgt: false
enc_num_layers: 2
encoder_hidden_size: 96
decoder_hidden_size: 96
arch: AutoEncoder
recurrent_arch_enc: StaticGraphGRU
recurrent_arch_decoder: StaticGraphGRU
diffusion_arch:
arch: Denoiser
use_attention: true
self_condition: false
norm_type: none
depth: 4
attn_dim_head: 32
attn_heads: 8
learn_influence: true
covariance_matrix_type: adjacency
hydra:
run:
dir: ./models
job:
chdir: true
|