| { | |
| "dataset_repo_id": "apaszynska/merged-recap-cup-training", | |
| "dataset_root": null, | |
| "video_backend": "pyav", | |
| "output_dir": "/workspace/outputs/pistar05_pretrain", | |
| "epochs": 15, | |
| "steps": null, | |
| "batch_size": 32, | |
| "lr": 0.001, | |
| "weight_decay": 0.01, | |
| "grad_clip_norm": 1.0, | |
| "warmup_steps": 500, | |
| "cfail": 100.0, | |
| "use_distributional": true, | |
| "num_value_bins": 201, | |
| "vlm_backbone": "google/gemma-3-270m", | |
| "freeze_vlm": true, | |
| "value_head_hidden_dim": 256, | |
| "value_head_num_layers": 2, | |
| "seed": 42, | |
| "device": "cuda", | |
| "num_workers": 8, | |
| "prefetch_factor": 2, | |
| "gpu_video_decode": false, | |
| "use_amp": true, | |
| "amp_dtype": "bfloat16", | |
| "log_freq": 200, | |
| "save_freq": 5, | |
| "wandb_project": null, | |
| "wandb_entity": null | |
| } |