github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

update docs (#4356)

Browse Source 
* update docs

* nn.md

* mnist cleanups

* rhip test is very slow
This commit is contained in:
George Hotz
2024-04-30 15:51:42 +08:00
committed by GitHub
parent a2d81514fd
commit d325be2540
9 changed files with 45 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
docs/nn.md
View File

@@ -1,17 +1,29 @@
## Neural Network classes
::: tinygrad.nn
options:
members: true
::: tinygrad.nn.BatchNorm2d
::: tinygrad.nn.Conv1d
::: tinygrad.nn.Conv2d
::: tinygrad.nn.ConvTranspose2d
::: tinygrad.nn.Linear
::: tinygrad.nn.GroupNorm
::: tinygrad.nn.InstanceNorm
::: tinygrad.nn.LayerNorm
::: tinygrad.nn.LayerNorm2d
::: tinygrad.nn.Embedding
## Optimizers
::: tinygrad.nn.optim
options:
members: true
::: tinygrad.nn.optim.SGD
::: tinygrad.nn.optim.LARS
::: tinygrad.nn.optim.AdamW
::: tinygrad.nn.optim.Adam
::: tinygrad.nn.optim.LAMB
## Load/Save
::: tinygrad.nn.state
options:
members: true
::: tinygrad.nn.state.safe_load
::: tinygrad.nn.state.safe_save
::: tinygrad.nn.state.get_state_dict
::: tinygrad.nn.state.get_parameters
::: tinygrad.nn.state.load_state_dict
::: tinygrad.nn.state.torch_load