Skip to content

Commit

Permalink
tip for integration with Pytorch Lightning
Browse files Browse the repository at this point in the history
  • Loading branch information
davda54 authored Jul 20, 2021
1 parent a001d45 commit 451432d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ for input, output in data:
```python
scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optimizer.base_optimizer, T_max=200)
```
- [@AlbertoSabater](https://github.com/AlbertoSabater): Integration with Pytorch Lightning — you can write the `training_step` function as:
- [@AlbertoSabater](https://github.com/AlbertoSabater): Integration with Pytorch Lightning — you can write the `training_step` function as:
```python
def training_step(self, batch, batch_idx):
optimizer = self.optimizers()
Expand Down

0 comments on commit 451432d

Please sign in to comment.