add device type
Browse files- config.json +1 -0
config.json
CHANGED
|
@@ -8,6 +8,7 @@
|
|
| 8 |
"2": "Wuchereria bancrofti",
|
| 9 |
"3": "Brugia malayi"
|
| 10 |
},
|
|
|
|
| 11 |
"layer_norm_eps": 0.001,
|
| 12 |
"threshold": 0.3,
|
| 13 |
"min_depth": 8,
|
|
|
|
| 8 |
"2": "Wuchereria bancrofti",
|
| 9 |
"3": "Brugia malayi"
|
| 10 |
},
|
| 11 |
+
"device": "GPU",
|
| 12 |
"layer_norm_eps": 0.001,
|
| 13 |
"threshold": 0.3,
|
| 14 |
"min_depth": 8,
|