Skip to content

Commit ffda903

Browse files
committed
update: visualize_optimizers
1 parent a268d8e commit ffda903

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

examples/visualize_optimizers.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,8 @@ def execute_experiments(
144144

145145

146146
def main():
147+
# referred https://github.com/jettify/pytorch-optimizer/blob/master/examples/viz_optimizers.py
148+
147149
np.random.seed(42)
148150
torch.manual_seed(42)
149151

@@ -180,3 +182,7 @@ def main():
180182
root_path,
181183
'rosenbrok',
182184
)
185+
186+
187+
if __name__ == '__main__':
188+
main()

0 commit comments

Comments
 (0)