File size: 1,159 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
defaults:
  - agent: dreamer
  - conf/env: dmc_pixels
  - conf/defaults: dreamer_v3
  - override hydra/launcher: submitit_local

# mode
label: default
# task settings
task: stickman_walk
# train settings
num_train_frames: 2000010
num_seed_frames: 4000
# eval
eval_every_frames: 100000
eval_modality: null
num_eval_episodes: 3
# snapshot
snapshot_dir: ../../../trained_models/${obs_type}/${task}/${agent.name}/${seed}
save_every_frames: 10_000
# misc
seed: 1
device: cuda:0
use_tb: true
use_wandb: true

# Clip stuff
viclip_encode: false
viclip_model: internvideo2
clip_hd_rendering: false

# experiment
experiment: data
project_name: genrl

# log settings
log_every_frames: 1000
visual_every_frames: 100000000 # edit for debug
workdir: ???

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