il_2.yaml

program: imitate_episodes_2.py
method: grid

parameters:
  task_name:
    # value: "sim_insertion_scripted"
    value: "sim_transfer_cube_scripted"
  policy_class:
    value: "ACT2"
  kl_weight:
    value: 10
  pred_weight:
    value: 0.005
  decay_rate:
    values: [1]
  chunk_size:
    values: [100]
  ckpt_dir:
    # value: "/localdata/yy/zzzzworkspace/act/ckpt/sim_transfer_cube_scripted_run3"
    # value: "./ckpt/sim_insertion_scripted/Testchunk"
    # value: "/localdata/yy/zzzzworkspace/act/ckpt/sim_insertion_human_chunk100"
    # value: "/localdata/yy/zzzzworkspace/act/ckpt/sim_transfer_cube_scripted_run2"
    value: "/home/torchuser/adaptiveAC/wandb_ckpt/vae_transfer_cube"
  prediction_ckpt_dir:
    value: "/home/torchuser/adaptiveAC/wandb_ckpt/vae_transfer_cube_prediction"
  hidden_dim:
    value: 512
  batch_size:
    value: 8
  dim_feedforward:
    value: 3200
  query_freq:
    values: [100]
  temporal_agg:
    value: false
  num_epochs:
    value: 2000
  num_epochs_prediction:
    value: 5
  lr:
    value: 1e-5
  lr_backbone:
    value: 1e-6
  lr_pred:
    value: 1e-6
  seed:
    values: [1]
  eval:
    value: false
  onscreen_render:
    value: false
  use_predict_model:
    value: false
  state_dim:
    value: 14
  action_dim:
    value: 14

command:
  - python3
  - ${program}
  - ${args_no_boolean_flags}