alexandersoare
commited on
Commit
•
eec666c
1
Parent(s):
4392425
Add use_amp for backwards compatibility
Browse files- config.yaml +1 -0
config.yaml
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
device: cuda
|
|
|
2 |
seed: 1000
|
3 |
dataset_repo_id: lerobot/aloha_sim_transfer_cube_human
|
4 |
training:
|
|
|
1 |
device: cuda
|
2 |
+
use_amp: false
|
3 |
seed: 1000
|
4 |
dataset_repo_id: lerobot/aloha_sim_transfer_cube_human
|
5 |
training:
|