Export fixes for Riva (#5496) (#5497)

* Export fixes for Riva Signed-off-by: Boris Fomitchev <[email protected]> * Cleaning up training_utils Signed-off-by: Boris Fomitchev <[email protected]> Signed-off-by: Boris Fomitchev <[email protected]> Signed-off-by: Boris Fomitchev <[email protected]> Co-authored-by: Boris Fomitchev <[email protected]>
NVIDIA · Nov 24, 2022 · 2430d0f · 2430d0f
1 parent b86bba8
commit 2430d0f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/utils/export_utils.py b/nemo/utils/export_utils.py
@@ -169,7 +169,7 @@ def run_ts_and_compare(ts_model, ts_input_list, ts_input_dict, output_example, c
 
         if torch.is_tensor(expected):
             tout = out.to('cpu')
-            logging.debug(f"Checking output {i}, shape: {expected.shape}:\n{expected}\n{tout}")
+            logging.debug(f"Checking output {i}, shape: {expected.shape}:\n")
             this_good = True
             try:
                 if not torch.allclose(tout, expected.cpu(), rtol=check_tolerance, atol=check_tolerance):
@@ -191,7 +191,7 @@ def run_ort_and_compare(sess, ort_input, output_example, check_tolerance=0.01):
 
         if torch.is_tensor(expected):
             tout = torch.from_numpy(out)
-            logging.debug(f"Checking output {i}, shape: {expected.shape}:\n{expected}\n{tout}")
+            logging.debug(f"Checking output {i}, shape: {expected.shape}:\n")
             this_good = True
             try:
                 if not torch.allclose(tout, expected.cpu(), rtol=check_tolerance, atol=100 * check_tolerance):