Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,696 Bytes
2d9a728 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
defaults:
- agent: genrl
- conf/defaults: genrl
- conf/env: dmc_pixels
- override hydra/launcher: submitit_local
# mode
label: default
# task settings
task: stickman_walk
# train settings
num_train_frames: 200_010
num_seed_frames: 400
# eval
eval_every_frames: 5_000
eval_modality: null
num_eval_episodes: 10
# snapshot
snapshot_dir: ../../../models/${obs_type}/${task}/${agent.name}/${seed}
snapshot_every_frames: 50_000
save_every_frames: 1000
# misc
seed: 1
device: cuda:0
use_tb: true
use_wandb: true
# Clip stuff
viclip_encode: true
viclip_model: internvideo2
clip_add_noise: 0.0
clip_hd_rendering: false
# experiment
experiment: train
project_name: genrl
# log settings
log_every_frames: 1_000
visual_every_frames: 100000000 # edit for debug
log_episodes_every_frames: 100_000
workdir: ???
# training models
train_connector: false
train_world_model: false
reset_connector: false
reset_world_model: false
reset_imag_behavior: true
# loading options
replay_from_wandb_project: null
snapshot_from_wandb_project: null
task_snapshot: null
seed_snapshot: null
agent_name_snapshot: null
snapshot_load_dir: null
replay_load_dir: null
#
snapshot_step: null
# data-free
train_from_data: true
start_from_video: mix
mix_random_actions: True
imag_warmup_steps: 5
hydra:
run:
dir: ./exp_local/${now:%Y.%m.%d}/${now:%H%M%S}_${agent.name}
sweep:
dir: ./exp_sweep/${now:%Y.%m.%d}/${now:%H%M}_${agent.name}_${experiment}
subdir: ${hydra.job.num}
launcher:
timeout_min: 4300
cpus_per_task: 10
gpus_per_node: 1
tasks_per_node: 1
mem_gb: 160
nodes: 1
submitit_folder: ./exp_sweep/${now:%Y.%m.%d}/${now:%H%M}_${agent.name}_${experiment}/.slurm
|