From 740cfa94078ca667142c3601b6d4434b2c9b9ddf Mon Sep 17 00:00:00 2001
From: piotrekobiIntel <piotr.paturej@intel.com>
Date: Mon, 21 Feb 2022 08:35:49 +0100
Subject: [PATCH] Add loss conversion from uint16 to float in ProgressBar class
 (#39231)

* Add loss conversion from uint16 to float in progressbar class

* Fix test coverage

* Actually fix coverage

* Fix format error
---
 python/paddle/hapi/progressbar.py       | 15 +++++++++++++++
 python/paddle/tests/test_progressbar.py |  1 +
 2 files changed, 16 insertions(+)

diff --git a/python/paddle/hapi/progressbar.py b/python/paddle/hapi/progressbar.py
index 6ed33f4f960..8020029be2a 100644
--- a/python/paddle/hapi/progressbar.py
+++ b/python/paddle/hapi/progressbar.py
@@ -20,6 +20,7 @@ import os
 import sys
 import time
 import numpy as np
+import struct
 from collections import namedtuple
 
 __all__ = []
@@ -79,6 +80,20 @@ class ProgressBar(object):
     def update(self, current_num, values={}):
         now = time.time()
 
+        def convert_uint16_to_float(in_list):
+            in_list = np.asarray(in_list)
+            out = np.vectorize(
+                lambda x: struct.unpack('<f', struct.pack('<I', x << 16))[0],
+                otypes=[np.float32])(in_list.flat)
+            return np.reshape(out, in_list.shape)
+
+        for i, (k, val) in enumerate(values):
+            if k == "loss":
+                val = val if isinstance(val, list) or isinstance(
+                    val, np.ndarray) else [val]
+                if isinstance(val[0], np.uint16):
+                    values[i] = ("loss", list(convert_uint16_to_float(val)))
+
         if current_num:
             time_per_unit = (now - self._start) / current_num
         else:
diff --git a/python/paddle/tests/test_progressbar.py b/python/paddle/tests/test_progressbar.py
index 47265229182..a68aee7aa8f 100644
--- a/python/paddle/tests/test_progressbar.py
+++ b/python/paddle/tests/test_progressbar.py
@@ -41,6 +41,7 @@ class TestProgressBar(unittest.TestCase):
         progbar.update(1, [['loss', 1e-4]])
         progbar.update(1, [['loss', np.array([1.])]])
         progbar.update(1, [['loss', np.array([1e-4])]])
+        progbar.update(1, [['loss', np.array([1]).astype(np.uint16)]])
         progbar.start()
 
         progbar.update(0, values)
-- 
GitLab