Skip to content

Commit b460d80

Browse files
committed
hardcode cpu backend
1 parent 1a90a3a commit b460d80

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/NeuralNetwork/index.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ class DiyNeuralNetwork {
113113
* @param {*} callback
114114
*/
115115
init(callback) {
116-
tf.setBackend("webgl");
116+
tf.setBackend("cpu");
117117
// check if the a static model should be built based on the inputs and output properties
118118
if (this.options.neuroEvolution === true) {
119119
this.createLayersNoTraining();

0 commit comments

Comments
 (0)