Hugging Face's logo

johnMinelli
/
CroSTAta_evaluation_checkpoints

Model card Files
xet
Community
CroSTAta_evaluation_checkpoints
File size: 2,319 Bytes 
d8f2c24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
88
89
90
91
92
93
94
95
96
97
98
99
 
seed: 42
device: cuda:0
n_epochs: 200
batch_size: 16
gradient_accumulation_steps: 1
mixed_precision: 'no'
learning_rate: 8.0e-05
lr_scheduler: constant
ent_coef: 0.01
l2_coef: 0.01
weight_decay: 0
max_grad_norm: 1.0
agent_class: sl_agent.SL
policy: TransformerPolicy
policy_kwargs:
  dim_decoder: 512
  n_decoder_layers: 4
  n_decoder_heads: 8
  use_rmsnorm: true
  pre_norm: true
  sa_flag: true
  ca_flag: true
  sta_flag: true
  sta_residual: false
  sta_norm: math
  dropout: 0.0
  dropout_residual: 0.0
  action_horizon: 16
  optimizer_kwargs:
    eps: 1.0e-08
  noise: true
preprocessor_class: Lerobot_2_Agent
preprocessor_kwargs:
  resize_images: !!python/tuple
  - 94
  - 94
  normalize_images: true
  observation_modes:
    observation.state: min_max
  action_modes:
    action: min_max
  overwrite_stats: false
  stats: null
  data_key_map:
    observation.images.base_camera: cam1
    observation.images.hand_camera: cam2
    observation.images.head_camera: cam2
    observation.images.image_side_1: cam1
    observation.images.image_side_2: cam2
    observation.images.image_wrist_1: cam3
    observation.images.image_wrist_2: cam4
    observation.top: cam1
    image: cam1
    wrist_image: cam2
    observation.state: state
    state: state
    action: action
    actions: action
  task: PegInsertionSide-v1
env_cfg:
  name: maniskill
  task: PegInsertionSide-v1
  dataset_repo_id: johnMinelli/ManiSkill_PegInsertionSide-v1_recovery
  episodes: null
  dataset_revision: v2.0
  dataset_root: null
  video_backend: pyav
  input_shapes:
    observation.images.base_camera: !!python/tuple
    - 480
    - 640
    - 3
    observation.images.hand_camera: !!python/tuple
    - 480
    - 640
    - 3
    observation.state: !!python/tuple
    - 9
    observation.privileged: !!python/tuple
    - 27
  output_shapes:
    action: !!python/tuple
    - 16
    - 8
  normalization:
    observation_modes:
      observation.state: min_max
    action_modes:
      action: min_max
  fps: 20
  delta_timestamps:
    expert_mask: '[i / 20 for i in range(16)]'
    observation.images.base_camera: '[i / 20 for i in range(16)]'
    observation.images.hand_camera: '[i / 20 for i in range(16)]'
    observation.state: '[i / 20 for i in range(16)]'
    action: '[i / 20 for i in range(16)]'
  image_transforms:
    enable: false