Skip to content

Commit

Permalink
Remove print
Browse files Browse the repository at this point in the history
  • Loading branch information
JulienVig committed Jul 1, 2024
1 parent 7095845 commit a730240
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion cli/src/benchmark_gpt.ts
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,6 @@ async function main(args: Required<CLIArguments>): Promise<void> {
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 <br> ${logs.peakMemory.toFixed(2)} GB`)
}

Expand Down

0 comments on commit a730240

Please sign in to comment.