y1y2y3 commited on
Commit
c5c2026
·
verified ·
1 Parent(s): d7de10d

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. train_config.json +2 -2
README.md CHANGED
@@ -6,8 +6,8 @@ model_name: diffusion
6
  pipeline_tag: robotics
7
  tags:
8
  - lerobot
9
- - robotics
10
  - diffusion
 
11
  ---
12
 
13
  # Model Card for diffusion
 
6
  pipeline_tag: robotics
7
  tags:
8
  - lerobot
 
9
  - diffusion
10
+ - robotics
11
  ---
12
 
13
  # Model Card for diffusion
config.json CHANGED
@@ -38,7 +38,7 @@
38
  ]
39
  }
40
  },
41
- "device": "cuda:1",
42
  "use_amp": false,
43
  "push_to_hub": true,
44
  "repo_id": "y1y2y3/so101_test4_diffusion_12k",
 
38
  ]
39
  }
40
  },
41
+ "device": "cuda:0",
42
  "use_amp": false,
43
  "push_to_hub": true,
44
  "repo_id": "y1y2y3/so101_test4_diffusion_12k",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:64e885291b9566ab5ef915846df240a5aa25d2bee919e43ae8f90753929916fb
3
  size 1066517744
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b195a81ed7a3be551c2290ca967580f8dadb52b78cd0f92ad960115cccf7c624
3
  size 1066517744
train_config.json CHANGED
@@ -105,7 +105,7 @@
105
  ]
106
  }
107
  },
108
- "device": "cuda:1",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "y1y2y3/so101_test4_diffusion_12k",
@@ -193,7 +193,7 @@
193
  "project": "lerobot",
194
  "entity": null,
195
  "notes": null,
196
- "run_id": null,
197
  "mode": null
198
  }
199
  }
 
105
  ]
106
  }
107
  },
108
+ "device": "cuda:0",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "y1y2y3/so101_test4_diffusion_12k",
 
193
  "project": "lerobot",
194
  "entity": null,
195
  "notes": null,
196
+ "run_id": "d6glqpex",
197
  "mode": null
198
  }
199
  }