diff --git a/python/paddle/hapi/callbacks.py b/python/paddle/hapi/callbacks.py index 61ae8b42d63a909cc4dc88d4b16f0b0e8ed83c71..834b92f9fe6a0c0abaa0946d1676029d4849ae45 100644 --- a/python/paddle/hapi/callbacks.py +++ b/python/paddle/hapi/callbacks.py @@ -395,6 +395,10 @@ class ProgBarLogger(Callback): values.append( ('ips', "%.5f samples/sec" % (samples / (timer['data_time'] + timer['batch_time'])))) + timer['count'] = 0 + timer['samples'] = 0 + timer['data_time'] = 0. + timer['batch_time'] = 0. progbar.update(steps, values)