在tf.max函数中面临内存泄漏

问题描述

async function predicttest() {
    while (true) {
        console.log('Executing');
        tf.engine().startScope();  
            const img = tf.ones([300,300])
            const fbmx = tf.tidy(() => img.max()); //giving memory leak,but min doesnt.
            img.dispose();
            fbmx.dispose();
        tf.engine().endScope();
        await tf.nextFrame(); 
      }
    }

在执行此while循环时（分配泄漏后隔离），分配的GPU内存继续增加。如果使用tf.min（）则没有问题。需要使用tf.max（）函数进行标准化，因此不可避免。

我尝试使用tf.setBackend（'webgl'）和tf.disposeVariables（），但无济于事。

如何解决此内存泄漏？

TFjs：2.3.0
浏览器：Google Chrome 74

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）