Function benchmarkModel

e2e/benchmarks/browserstack-benchmark/benchmark_models.js:61–89

(benchmarkParameters)

const KARMA_SERVER = './base';

async function benchmarkModel(benchmarkParameters) {
  // Load the model.
  const benchmark = benchmarks[benchmarkParameters.model];
  const numRuns = benchmarkParameters.numRuns;
  let model;
  if (benchmarkParameters.model === 'custom') {
    if (benchmarkParameters.modelUrl == null) {
      throw new Error('Please provide model url for the custom model.');
    }
    model = await loadModelByUrl(benchmarkParameters.modelUrl);
  } else {
    model = await benchmark.load();
  }

  // Benchmark.
  let timeInfo;
  let memoryInfo;
  if (benchmark.predictFunc != null) {
    const predict = benchmark.predictFunc();
    timeInfo = await timeInference(() => predict(model), numRuns);
    memoryInfo = await profileInference(() => predict(model));
  } else {
    const input = generateInput(model);
    timeInfo = await timeModelInference(model, input, numRuns);
    memoryInfo = await profileModelInference(model, input);
  }

  return { timeInfo, memoryInfo };
}

async function benchmarkCodeSnippet(benchmarkParameters) {
  let predict = null;