Update config.yaml
Browse files- config.yaml +1 -1
config.yaml
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
resume: false
|
| 2 |
-
device:
|
| 3 |
use_amp: false
|
| 4 |
seed: 1000
|
| 5 |
dataset_repo_id: danmac1/red_ball
|
|
|
|
| 1 |
resume: false
|
| 2 |
+
device: cpu
|
| 3 |
use_amp: false
|
| 4 |
seed: 1000
|
| 5 |
dataset_repo_id: danmac1/red_ball
|