Skip to content

Commit

Permalink
Fix requirements-extra.txt Trains package to release version (#1229)
Browse files Browse the repository at this point in the history
* Fix requirement-extra use released Trains package

* Update README.md add Trains and links to the external Visualization section

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
  • Loading branch information
bmartinn and Borda authored Mar 27, 2020
1 parent d394b80 commit 6a0b171
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 5 deletions.
10 changes: 6 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -233,10 +233,12 @@ trainer.test()
## Visualization
Lightning has out-of-the-box integration with the popular logging/visualizing frameworks

- Tensorboard
- MLFlow
- Neptune.ai
- Comet.ml
- [Tensorboard](https://pytorch.org/docs/stable/tensorboard.html)
- [MLFlow](https://mlflow.org/)
- [Neptune.ai](https://neptune.ai/)
- [Comet.ml](https://www.comet.ml/site/)
- [Wandb](https://www.wandb.com/)
- [Trains](https://github.com/allegroai/trains)
- ...

![tensorboard-support](docs/source/_images/general/tf_loss.png)
Expand Down
2 changes: 1 addition & 1 deletion requirements-extra.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,4 @@ comet-ml>=1.0.56
mlflow>=1.0.0
test_tube>=0.7.5
wandb>=0.8.21
trains>=0.14.1rc0
trains>=0.14.1

0 comments on commit 6a0b171

Please sign in to comment.