fix memory

Author: Mmasoud1

js/brainchop/mainMeshNetFunctions.js

@@ -4141,8 +4141,7 @@ class SequentialConvLayer {
         // Important to avoid "undefined" class var members inside the timer.
         // "this" has another meaning inside the timer.
 
-        // *** WARNING!!! if you uncomment this line the memory leak will break webGL and may reboot your machine
-        //document.getElementById("progressBarChild").parentElement.style.visibility = "visible";
+        document.getElementById("progressBarChild").parentElement.style.visibility = "visible";
 
         return new Promise((resolve, reject) => {
 
@@ -4184,32 +4183,27 @@ class SequentialConvLayer {
                   });
 
                   // -- await showMemStatus(chIdx, self.outChannels);
-                  // Log memory usage
-                     const memoryInfo = tf.memory();
-                     console.log(`Iteration ${chIdx}:`);
-                     console.log(`Number of Tensors: ${memoryInfo.numTensors}`);
-                     console.log(`Number of Data Buffers: ${memoryInfo.numDataBuffers}`);
-                     console.log(`Bytes In Use: ${memoryInfo.numBytes}`);
-                     console.log(`Megabytes In Use: ${(memoryInfo.numBytes / 1048576).toFixed(3)} MB`);
-                     console.log(`Unreliable: ${memoryInfo.unreliable}`);
+
                   // Assign the new values to outC and outB
                   outC = result[0];
                   outB = result[1];
 
                   if(chIdx == (self.outChannels -1)) {
 
                       window.clearInterval( seqTimer );
-        // *** WARNING!!! if you uncomment this line the memory leak will break webGL and may reboot your machine
-                      // document.getElementById("progressBarChild").style.width = 0 + "%";
+                      document.getElementById("progressBarChild").style.width = 0 + "%";
                       tf.dispose(outB);
                       const endTime = performance.now();
                       const executionTime = endTime - startTime;
                       console.log(`Execution time for output layer: ${executionTime} milliseconds`);
                       resolve(outC);
+                  } else {
+
+                    chIdx++;
+                    document.getElementById("progressBarChild").style.width = (chIdx + 1)*100/self.outChannels + "%";
                   }
-		  chIdx++;
-        // *** WARNING!!! if you uncomment this line the memory leak will break webGL and may reboot your machine
-                  //document.getElementById("progressBarChild").style.width = (chIdx + 1)*100/self.outChannels + "%";
+
+
               }, 10);
         });