wozeparrot 67de3aa1de Add mlperf bert model (#803) ...

* feat: add mlperf bert model

* feat: switch to nn.Embedding

* clean+fix: fix formatting

* feat: add simple downloader

* feat: metrics

* feat: don't actually need exact match

* feat: doing a run

* feat: set eps on the layernorms

* clean+fix: cleaner impl + hopefully fixed

* feat: move dataset initialization into iterate

* feat: move tokenizer out of iterate

* clean+fix: cleaner + working

* clean: cleanup

* fix: fix metrics

* feat: need to use original bert gelu + download vocab

* feat: make directory if it doesn't exist yet

* feat: jit go brrr

2023-05-27 14:53:32 -07:00

..

bert.py

Add mlperf bert model (#803)

2023-05-27 14:53:32 -07:00

convnext.py

LayerNorm2d for 2 lines

2023-03-20 16:58:43 -07:00

efficientnet.py

third try at torch loading (#677)

2023-03-10 19:11:29 -08:00

resnet.py

getting 77% on imagenet eval

2023-05-13 07:46:27 -07:00

rnnt.py

Promote Embedding to nn (#798)

2023-05-25 18:39:45 -07:00

transformer.py

good changes from llama branch (#671)

2023-03-09 20:51:22 -08:00

unet3d.py

conv3d + unet3d (#772)

2023-05-12 13:54:07 -07:00

vit.py

good changes from llama branch (#671)

2023-03-09 20:51:22 -08:00