nit

Author: NihalHarish

smdebug/pytorch/hook.py

@@ -156,7 +156,7 @@ def forward_hook(self, module, inputs, outputs):
             return
 
         if isinstance(module, _Loss):
-            module_name = module._get_name()
+            module._module_name = module._get_name()
 
         module_name = module._module_name
         # This overwhelms the logs; turn back on if you really need it