Correctly log class that was restored

Signed-off-by: smajumdar <titu1994@gmail.com>
NVIDIA · Jun 11, 2021 · 80fd773 · 80fd773
1 parent 2133f51
commit 80fd773
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/core/classes/modelPT.py b/nemo/core/classes/modelPT.py
@@ -436,7 +436,7 @@ def _default_restore_from(
                     instance = instance.to(map_location)
                     instance.load_state_dict(torch.load(model_weights, map_location=map_location), strict=strict)
 
-                    logging.info(f'Model {cls.__name__} was successfully restored from {restore_path}.')
+                    logging.info(f'Model {instance.__class__.__name__} was successfully restored from {restore_path}.')
             finally:
                 cls._set_model_restore_state(is_being_restored=False)
                 os.chdir(cwd)