Lerobot/docker/train_task.json

12 lines
329 B
JSON

{
"task_id": "b5c75014c1142feab3ee395b4a0bcc0",
"gpu_id": "0",
"train": {
"model":"smolvla",
"epochs":20,
"batch_size":64,
"log_path": "/workspace/logs",
"checkpoint_path": "/workspace/outputs/checkpoints",
"input_data_path": "/workspace/inputs/pick_red_marker"
}
}