more formatting, ready for new version

experiments/paper_demo.py

@@ -27,7 +27,7 @@
 
 # repeat the experiment multiple times (generates multiple files to be aggregated by `get_best_results`)
 n_repeat = 5
-batchnorm_eval = True # BatchNorm in eval mode
+batchnorm_eval = True  # BatchNorm in eval mode
 
 # ============== CONV CONFIG ==============
 # Valid choices for models are in models.conv_model_fns

experiments/util/estimate.py

@@ -80,13 +80,17 @@ def skip_case_check(args: argparse.Namespace) -> bool:
     invalid = False
     if args.case is None:
         return invalid
-    is_surgical = 'surgical_last' in args.case or 'surgical_first' in args.case
+    is_surgical = "surgical_last" in args.case or "surgical_first" in args.case
     # 1.
     for c in ["grad_norm_bias", "grad_norm_weights"]:
         if c in args.case and args.model in models.models_without_norm:
             invalid = True
     for c in ["no_grad_norm_bias", "no_grad_norm_weights"]:
-        if c not in args.case and args.model in models.models_without_norm and not is_surgical:
+        if (
+            c not in args.case
+            and args.model in models.models_without_norm
+            and not is_surgical
+        ):
             invalid = True
     # 2.
     if "no_grad_embed_weights" in args.case and "grad_input" not in args.case:

memsave_torch/nn/Dropout.py

@@ -42,4 +42,3 @@ def from_nn_dropout(cls, dropout: nn.Dropout):
         """
         obj = cls(dropout.p)
         return obj
-