From 971c8301a93a54b63740685bc359793af98a0fd2 Mon Sep 17 00:00:00 2001
From: Mattk70 <email@mattkirkland.co.uk>
Date: Mon, 15 Apr 2024 13:11:23 +0100
Subject: [PATCH] Enabled oneDNN optimisations bumped version

---
 js/BirdNet2.4.js | 2 +-
 main.js          | 1 +
 package.json     | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/js/BirdNet2.4.js b/js/BirdNet2.4.js
index 889d1fac..7e840caa 100644
--- 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
index 42c95840..b873a247 100644
--- 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
index 925a90d2..140c67ba 100644
--- 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": {