k1000dai commited on
Commit
f8484ce
·
verified ·
1 Parent(s): d20b0b8

Update train_config.json

Browse files
Files changed (1) hide show
  1. train_config.json +1 -1
train_config.json CHANGED
@@ -105,7 +105,7 @@
105
  ]
106
  }
107
  },
108
- "device": "cuda:0",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "k1000dai/internvla_libero",
 
105
  ]
106
  }
107
  },
108
+ "device": "cuda",
109
  "use_amp": false,
110
  "push_to_hub": true,
111
  "repo_id": "k1000dai/internvla_libero",