anygoal / driftworld_insertpeg.json
HazelNam's picture
Upload driftworld_insertpeg.json with huggingface_hub
3da5d7d verified
Raw History Blame Contribute Delete
690 Bytes
{
"dataset": "insertpeg",
"format_version": 1,
"frame_skip": 5,
"learning_rate": 5e-05,
"model": "driftworld",
"sha256": "34d9a8655eb62c5f7144a47aeeb74f7e98ad1fcfea5210e3829c1c337d700dec",
"size_bytes": 1187191479,
"source_config_filename": "config.yaml",
"source_filename": "ckpt-step100000.pth",
"training_config": {
"data.batch_size": 2,
"data.frameskip": 5,
"model.action_dim": 5,
"opt.lr": 5e-05,
"opt.weight_decay": 0.01,
"train.max_steps": 100000,
"train.num_epochs": 300,
"train.seed": 1,
"visualization.seed": 1
},
"training_run": "20260910_driftworld_manipulator_insert_peg_lewm_fs5_lr5e-5",
"training_step": 100000
}