Hugging Face's logo

tmeynier
/
hepha-diffusion-200-simple-drawer-5-obs4

Model card Files
xet
 Community
hepha-diffusion-200-simple-drawer-5-obs4
File size: 2,277 Bytes
52eaaf6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "args": {
    "dataset_dir": "datasets/hepha_act_200_simple_drawer_5",
    "output_dir": "runs/hepha_diffusion_200_simple_drawer_5_obs4",
    "epochs": 50,
    "batch_size": 16,
    "num_workers": 4,
    "lr": 0.0001,
    "weight_decay": 1e-06,
    "chunk_size": 50,
    "image_height": 240,
    "image_width": 320,
    "obs_feature_dim": 512,
    "n_obs_steps": 4,
    "n_action_steps": null,
    "diffusion_step_embed_dim": 256,
    "down_dims": [
      256,
      512,
      1024
    ],
    "kernel_size": 5,
    "n_groups": 8,
    "cond_predict_scale": true,
    "diffusion_steps": 100,
    "num_inference_steps": 100,
    "beta_schedule": "squaredcos_cap_v2",
    "prediction_type": "epsilon",
    "backbone": "resnet18",
    "lr_backbone": 1e-05,
    "no_pretrained_backbone": false,
    "dropout": 0.1,
    "val_ratio": 0.1,
    "val_every_steps": 4000,
    "seed": 42,
    "log_every": 50,
    "save_every": 0,
    "save_every_steps": 4000,
    "resume": null,
    "amp": true,
    "device": "cuda",
    "wandb_project": "hepha-diffusion",
    "wandb_entity": null,
    "wandb_name": "hepha-diffusion-200-simple-drawer-5-obs4",
    "wandb_tags": null,
    "wandb_mode": null,
    "wandb_log_checkpoints": false,
    "hf_model_repo_id": "tmeynier/hepha-diffusion-200-simple-drawer-5-obs4",
    "hf_revision": null,
    "hf_private": true,
    "hf_upload_every": null,
    "hf_upload_every_steps": 4000,
    "hf_upload_fail_training": false
  },
  "model_config": {
    "state_dim": 16,
    "action_dim": 15,
    "chunk_size": 50,
    "horizon": 50,
    "n_obs_steps": 4,
    "n_action_steps": 50,
    "obs_feature_dim": 512,
    "diffusion_step_embed_dim": 256,
    "down_dims": [
      256,
      512,
      1024
    ],
    "kernel_size": 5,
    "n_groups": 8,
    "cond_predict_scale": true,
    "diffusion_steps": 100,
    "num_inference_steps": 100,
    "beta_start": 0.0001,
    "beta_end": 0.02,
    "beta_schedule": "squaredcos_cap_v2",
    "prediction_type": "epsilon",
    "clip_sample": true,
    "dropout": 0.1,
    "backbone": "resnet18",
    "pretrained_backbone": true
  },
  "dataset_info": {
    "dataset_dir": "datasets/hepha_act_200_simple_drawer_5",
    "total_frames": 253867,
    "episodes": 200,
    "state_dim": 16,
    "action_dim": 15
  }
}