Skip to content
Snippets Groups Projects

Allow specifying device to use when training

Merged Allow specifying device to use when training
export-device-when-training into main
All threads resolved!
Merged Manon Blanco requested to merge export-device-when-training into main
All threads resolved!
Viewing commit 69c8df89
Next
Show latest version
5 files
+ 41
− 42
Compare changes
  • Side-by-side
  • Inline
Files
5
configs/quickstart.json
+ 1
− 1
@@ -71,7 +71,7 @@
"ddp_port": "20027",
"use_amp": true,
"nb_gpu": null,
"force_cpu": false
"force": null
},
"metrics": {
"train": [