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