From b036ecb1649840fcc13474b8b183146fba454c2d Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 12 Jan 2023 13:35:08 +0800 Subject: [PATCH] add numpy version (#7614) --- ppdet/engine/callbacks.py | 5 ++--- requirements.txt | 1 + 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ppdet/engine/callbacks.py b/ppdet/engine/callbacks.py index 3a644b1df..ece2ca8b3 100644 --- a/ppdet/engine/callbacks.py +++ b/ppdet/engine/callbacks.py @@ -152,7 +152,7 @@ class LogPrinter(Callback): if mode == 'eval': sample_num = status['sample_num'] cost_time = status['cost_time'] - logger.info('Total sample number: {}, averge FPS: {}'.format( + logger.info('Total sample number: {}, average FPS: {}'.format( sample_num, sample_num / cost_time)) @@ -369,7 +369,7 @@ class WandbCallback(Callback): metadata["last_epoch"] = last_epoch if ap: metadata["ap"] = ap - + if fps: metadata["fps"] = fps @@ -417,7 +417,6 @@ class WandbCallback(Callback): metrics = {"train/" + k: v for k, v in training_status.items()} - metrics["train/ips"] = ips metrics["train/data_cost"] = data_cost metrics["train/batch_cost"] = batch_cost diff --git a/requirements.txt b/requirements.txt index e789157f2..0cdbf8e22 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ +numpy < 1.24 tqdm typeguard visualdl>=2.2.0 -- GitLab