Skip to content

Commit

Permalink
delete example to pass flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
VCasecnikovs committed Jul 20, 2020
1 parent f7d4257 commit 00f457a
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions lars.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,6 @@ class LARS(Optimizer):
Based on Algorithm 1 of the following paper by You, Gitman, and Ginsburg.
Large Batch Training of Convolutional Networks:
https://arxiv.org/abs/1708.03888
Example:
>>> optimizer = LARS(model.parameters(), lr=0.1, eta=1e-3)
>>> optimizer.zero_grad()
>>> loss_fn(model(input), target).backward()
>>> optimizer.step()
"""
def __init__(self, params, lr=required, momentum=.9,
weight_decay=.0005, eta=0.001, max_epoch=200):
Expand Down

0 comments on commit 00f457a

Please sign in to comment.