Enabled oneDNN optimisations

bumped version
Mattk70 · Apr 15, 2024 · 971c830 · 971c830
1 parent 1b6fcd6
commit 971c830
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/js/BirdNet2.4.js b/js/BirdNet2.4.js
@@ -40,7 +40,7 @@ onmessage = async (e) => {
                 DEBUG && console.log(`model received load instruction. Using list: ${list}, batch size ${batch}`);
 
                 tf.setBackend('tensorflow').then(async () => {
-                    tf.enableProdMode();
+                    //tf.enableProdMode();
                     if (DEBUG) {
                         console.log(tf.env());
                         console.log(tf.env().getFlags());

diff --git a/main.js b/main.js
@@ -5,6 +5,7 @@ const log = require('electron-log');
 const fs = require("node:fs");
 const path = require('node:path');
 const settings = require('electron-settings');
+process.env['TF_ENABLE_ONEDNN_OPTS'] = "1";
 
 //require('update-electron-app')();
 let files = [];

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "chirpity",
-  "version": "1.6.10",
+  "version": "1.7.0",
   "description": "Chirpity Nocmig",
   "main": "main.js",
   "scripts": {