diff --git a/timm/utils.py b/timm/utils.py index 626ae9dc..88de12ef 100644 --- a/timm/utils.py +++ b/timm/utils.py @@ -18,7 +18,7 @@ def get_state_dict(model): if isinstance(model, ModelEma): return get_state_dict(model.ema) else: - return model.module.state_dict() if getattr(model, 'module') else model.state_dict() + return model.module.state_dict() if hasattr(model, 'module') else model.state_dict() class CheckpointSaver: