fix: replace bare except clauses with except Exception

Bare `except:` catches BaseException including KeyboardInterrupt and
SystemExit. Replaced 5 instances with `except Exception:`.

Author: haosenwang1018

tests/test_models.py

@@ -190,7 +190,7 @@ def test_model_forward(model_name, batch_size):
     # Test that grad-checkpointing, if supported, doesn't cause model failures or change in output
     try:
         model.set_grad_checkpointing()
-    except:
+    except Exception:
         # throws if not supported, that's fine
         pass
     else:
@@ -551,7 +551,7 @@ def test_model_forward_intermediates(model_name, batch_size):
     # Test that grad-checkpointing, if supported
     try:
         model.set_grad_checkpointing()
-    except:
+    except Exception:
         # throws if not supported, that's fine
         pass
     else:

timm/optim/adamw.py

@@ -205,7 +205,7 @@ def adamw(
         try:
             # cannot do foreach if this overload doesn't exist when caution enabled
             foreach = not caution or 'Scalar' in torch.ops.aten._foreach_maximum_.overloads()
-        except:
+        except Exception:
             foreach = False
 
     if foreach and not torch.jit.is_scripting():

timm/optim/adan.py

@@ -124,7 +124,7 @@ def step(self, closure=None):
 
         try:
             has_scalar_maximum = 'Scalar' in torch.ops.aten._foreach_maximum_.overloads()
-        except:
+        except Exception:
             has_scalar_maximum = False
 
         for group in self.param_groups:

timm/optim/nadamw.py

@@ -182,7 +182,7 @@ def nadamw(
         try:
             # cannot do foreach if this overload doesn't exist when caution enabled
             foreach = not caution or 'Scalar' in torch.ops.aten._foreach_maximum_.overloads()
-        except:
+        except Exception:
             foreach = False
 
     if foreach and not torch.jit.is_scripting():