File size: 1,692 Bytes
d809325 | 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 | {
"args": {
"dataset": "/kaggle/input/datasets/ernestbeckham/lafan1/bvh",
"dataset_legacy": "",
"dataset_legacy2": "",
"pattern": "walk*,run*,sprint*",
"fps": 60.0,
"max_frames": 0,
"full_skeleton": false,
"rebuild_database": false,
"out": "",
"work_dir": "",
"database_dir": "",
"cache_dir": "",
"run_name": "phase1",
"window": 64,
"stride": 32,
"batch_size": 16,
"validation_fraction": 0.15,
"workers": 0,
"backbone": "mamba",
"d_model": 256,
"n_layers": 4,
"d_state": 16,
"head": "full",
"no_contacts": false,
"input_dropout": 0.0,
"hidden_dropout": 0.0,
"use_retrieval": false,
"k": 4,
"retrieval_dropout": 0.1,
"no_leave_one_clip_out": false,
"epochs": 50,
"learning_rate": 0.001,
"weight_decay": 0.0001,
"grad_clip": 1.0,
"lr_schedule": "cosine",
"warmup_epochs": 1.0,
"min_lr_factor": 0.05,
"ss_start": 2,
"ss_end": 15,
"ss_max": 0.8,
"patience": 10,
"monitor": "val_rollout_total",
"monitor_start_epoch": 0,
"rollout_eval_batches": 8,
"residual_weight": 0.0,
"compile_policy": false,
"lr_scale": "linear",
"device": null,
"seed": 0,
"no_tensorboard": false,
"progress_every": 25,
"resume": "auto",
"hf_repo": "ErnestBeckham/mmm",
"hf_public": false,
"hf_license": ""
},
"workspace": {
"platform": "kaggle",
"root": "/kaggle/working",
"out": "/kaggle/working/runs/phase1",
"database": "/kaggle/working/data/databases/phase1",
"cache": "/kaggle/working/data/cache"
},
"world_size": 2,
"effective_batch": 32,
"learning_rate": 0.002
} |