Fix exception type being caught in checkpoint load sanity check

distiller/apputils/checkpoint.py

@@ -178,7 +178,7 @@ def load_checkpoint(model, chkpt_file, optimizer=None,
         try:
             if model.arch != checkpoint["arch"]:
                 raise ValueError("The model architecture does not match the checkpoint architecture")
-        except (NameError, KeyError):
+        except (AttributeError, KeyError):
             # One of the values is missing so we can't perform the comparison
             pass