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

..

mnist

move datasets into datasets

2021-10-30 19:55:50 -07:00

__init__.py

start on mlperf models

2023-05-10 16:30:49 -07:00

imagenet.py

Cleanup mlperf (#797)

2023-05-25 11:36:43 -07:00

librispeech.py

Cleanup mlperf (#797)

2023-05-25 11:36:43 -07:00

squad.py

Add mlperf bert model (#803)

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