Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
William Falcon 2021-02-13 14:41:38 -05:00 committed by GitHub
parent d924dd6a41
commit 194f048263
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
README.md
View File

@ -213,9 +213,12 @@ Lightning has over [40+ advanced features](https://pytorch-lightning.readthedocs
Here are some examples:
<div align="center">
<img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/features_2.jpg" max-height="600px">
</div>
<details>
<summary>Train on GPUs without code changes</summary>
<summary>Highlighted feature code snippets</summary>
```python
# 8 GPUs
@ -225,27 +228,21 @@ Here are some examples:
# 256 GPUs
trainer = Trainer(max_epochs=1, gpus=8, num_nodes=32)
```
</details>
<details>
<summary>Train on TPUs without code changes</summary>
```python
# no code changes needed
trainer = Trainer(tpu_cores=8)
```
</details>
<details>
<summary>16-bit precision</summary>
```python
# no code changes needed
trainer = Trainer(precision=16)
```
</details>
<details>
<summary>Experiment managers</summary>
```python
@ -268,27 +265,21 @@ Here are some examples:
# ... and dozens more
```
</details>
<details>
<summary>EarlyStopping</summary>
```python
es = EarlyStopping(monitor='val_loss')
trainer = Trainer(callbacks=[es])
```
</details>
<details>
<summary>Checkpointing</summary>
```python
checkpointing = ModelCheckpoint(monitor='val_loss')
trainer = Trainer(callbacks=[checkpointing])
```
</details>
<details>
<summary>Export to torchscript (JIT) (production use)</summary>
```python
@ -296,9 +287,7 @@ Here are some examples:
autoencoder = LitAutoEncoder()
torch.jit.save(autoencoder.to_torchscript(), "model.pt")
```
</details>
<details>
<summary>Export to ONNX (production use)</summary>
```python