config.json

{
    "task_name": "sim_gearbox_assembly_demos_filtered",
    "ckpt_dir": "/root/gearboxAssembly/data/model",
    "policy_class": "ACT",
    "kl_weight": 10,
    "chunk_size": 100,
    "hidden_dim": 512,
    "batch_size": 32,
    "dim_feedforward": 3200,
    "num_epochs": 10000,
    "lr": 1e-05,
    "seed": 0,
    "save_interval": 200,
    "early_stopping_patience": 3000
}