update: visualize_optimizers

kozistr · Jul 7, 2024 · ffda903 · ffda903
1 parent a268d8e
commit ffda903
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/visualize_optimizers.py b/examples/visualize_optimizers.py
@@ -144,6 +144,8 @@ def execute_experiments(
 
 
 def main():
+    # referred https://github.com/jettify/pytorch-optimizer/blob/master/examples/viz_optimizers.py
+
     np.random.seed(42)
     torch.manual_seed(42)
 
@@ -180,3 +182,7 @@ def main():
         root_path,
         'rosenbrok',
     )
+
+
+if __name__ == '__main__':
+    main()