wzn12's picture
Upload config.json with huggingface_hub
d034850 verified
Raw
History Blame Contribute Delete
777 Bytes
{
"job_name": "critic_warmup_smolvla_ring",
"output_dir": "outputs/train/critic_warmup_smolvla_ring",
"seed": 42,
"dataset_repo_id": "wzn12/teleop_ring_labeled",
"max_samples": 4000,
"train_test_split": 0.8,
"batch_size": 48,
"epochs": 20,
"num_workers": 4,
"learning_rate": 0.0003,
"weight_decay": 0.0001,
"grad_clip_norm": 1.0,
"scheduler_step_size": 2,
"scheduler_gamma": 0.9,
"log_freq": 10,
"save_freq": 1,
"eval_freq": 2,
"wandb_enable": true,
"wandb_project": "lerobot_critic_warmup",
"wandb_entity": null,
"wandb_notes": "Critic offline warmup training for SmolVLA-SAC policy",
"wandb_tags": [
"critic",
"warmup",
"smolvla",
"ring"
],
"push_to_hub": true,
"repo_id": "wzn12/critic_warmup_smolvla_ring"
}