@vladmandic/face-api/src/common/disposeUnusedWeightTensors.ts

JavaScript module for Face Detection and Face Recognition Using Tensorflow/JS

import { ParamMapping } from './types';

export function disposeUnusedWeightTensors(weightMap: any, paramMappings: ParamMapping[]) {
  Object.keys(weightMap).forEach(path => {
    if (!paramMappings.some(pm => pm.originalPath === path)) {
      weightMap[path].dispose()
    }
  })
}