Spaces:
Running
Running
| task_name: train | |
| tags: | |
| - dev | |
| train: true | |
| test: false | |
| ckpt_path: null | |
| seed: null | |
| trainer: | |
| _target_: pytorch_lightning.Trainer | |
| default_root_dir: ${paths.output_dir} | |
| accelerator: gpu | |
| devices: 1 | |
| deterministic: false | |
| num_sanity_val_steps: 0 | |
| log_every_n_steps: ${GENERAL.LOG_STEPS} | |
| val_check_interval: ${GENERAL.VAL_STEPS} | |
| check_val_every_n_epoch: ${GENERAL.VAL_EPOCHS} | |
| precision: 16-mixed | |
| max_steps: ${GENERAL.TOTAL_STEPS} | |
| limit_val_batches: 80 | |
| paths: | |
| root_dir: ${oc.env:PROJECT_ROOT} | |
| data_dir: ${paths.root_dir}/data/ | |
| log_dir: logs/ | |
| output_dir: ${hydra:runtime.output_dir} | |
| work_dir: ${hydra:runtime.cwd} | |
| extras: | |
| ignore_warnings: false | |
| enforce_tags: true | |
| print_config: true | |
| exp_name: AniMer | |
| SMAL: | |
| MODEL_PATH: ./data/my_smpl_00781_4_all.pkl | |
| NUM_JOINTS: 34 | |
| EXTRA: | |
| FOCAL_LENGTH: 1000 | |
| NUM_LOG_IMAGES: 4 | |
| NUM_LOG_SAMPLES_PER_IMAGE: 4 | |
| PELVIS_IND: 0 | |
| MODEL: | |
| USE_CLS: true | |
| IMAGE_SIZE: 256 | |
| IMAGE_MEAN: | |
| - 0.485 | |
| - 0.456 | |
| - 0.406 | |
| IMAGE_STD: | |
| - 0.229 | |
| - 0.224 | |
| - 0.225 | |
| BACKBONE: | |
| TYPE: vit | |
| SMAL_HEAD: | |
| TYPE: transformer_decoder | |
| IN_CHANNELS: 2048 | |
| IEF_ITERS: 1 | |
| TRANSFORMER_DECODER: | |
| depth: 6 | |
| heads: 8 | |
| mlp_dim: 1024 | |
| dim_head: 64 | |
| dropout: 0.0 | |
| emb_dropout: 0.0 | |
| norm: layer | |
| context_dim: 1280 | |