From a73024056188d339de666cdf9659bd64b7173591 Mon Sep 17 00:00:00 2001 From: Julien Vignoud Date: Mon, 1 Jul 2024 12:08:56 +0200 Subject: [PATCH] Remove print --- cli/src/benchmark_gpt.ts | 1 - 1 file changed, 1 deletion(-) diff --git a/cli/src/benchmark_gpt.ts b/cli/src/benchmark_gpt.ts index 8fe6dd165..1a381c890 100644 --- a/cli/src/benchmark_gpt.ts +++ b/cli/src/benchmark_gpt.ts @@ -84,7 +84,6 @@ async function main(args: Required): Promise { const [_, logs] = await async_iterator.gather(model.train(preprocessedDataset)) epochTime = (performance.now() - epochTime) const msPerToken = epochTime / (batchSize * contextLength * iterationsPerEpoch * epochsCounter) - console.log(epochTime, batchSize, contextLength ,iterationsPerEpoch ,epochsCounter) console.log(`\t\tTraining time: ${msPerToken.toFixed(2)} ms/token
${logs.peakMemory.toFixed(2)} GB`) }