Style

Author: addtt

Diff for: evaluate.py

@@ -70,15 +70,14 @@ def _add_args(self, parser: argparse.ArgumentParser) -> None:
                             dest='prior_samples',
                             metavar='N',
                             help="number of batches of samples from prior")
-        parser.add_argument(
-            '--layer-repr',
-            action='store_true',
-            dest='inspect_layer_repr',
-            help='inspect layer representations. Generate samples '
-            'by sampling top layers once, then taking many '
-            'samples from a middle layer, and finally sample '
-            'the downstream layers from the conditional mode. '
-            'Do this for every layer.')
+        parser.add_argument('--layer-repr',
+                            action='store_true',
+                            dest='inspect_layer_repr',
+                            help='inspect layer representations. Generate '
+                            'samples by sampling top layers once, then taking '
+                            'many samples from a middle layer, and finally '
+                            'sample the downstream layers from the conditional '
+                            'mode. Do this for every layer.')
 
     @classmethod
     def _check_args(cls, args: argparse.Namespace) -> argparse.Namespace:
@@ -93,11 +92,6 @@ def _check_args(cls, args: argparse.Namespace) -> argparse.Namespace:
         return args
 
 
-def main():
-    evaluator = Evaluator(experiment_class=LVAEExperiment)
-    evaluator()
-
-
 def inspect_layer_repr(model, img_folder, n=8):
     for i in range(model.n_layers):
 
@@ -120,5 +114,10 @@ def inspect_layer_repr(model, img_folder, n=8):
         save_image(sample, fname, nrow=n, pad_value=pad_value)
 
 
+def main():
+    evaluator = Evaluator(experiment_class=LVAEExperiment)
+    evaluator()
+
+
 if __name__ == "__main__":
     main()