Elias Wahl 27613dd881 MLPerf BERT: Main training loop (#4288) ... * BERT language modeling head + trunc normal initializers * add train loop + helpers * shuffle in dataloaders + slight changes in main loop * beam change * Minor changes * random.shuffle * HParam update * Use deque for dataloader * wandb bert project name * half fixes * BENCHMARK + remove epoch * cast + print() --------- Co-authored-by: chenyu <chenyu@fastmail.com>		2024-04-29 14:35:27 -04:00
..
bert.py	MLPerf BERT: Main training loop (#4288)	2024-04-29 14:35:27 -04:00
convnext.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
efficientnet.py	simple LoadOps.ASSIGN (#3745)	2024-03-14 20:44:34 -07:00
llama.py	hotfix: revert llama change	2024-04-10 20:13:15 -07:00
mask_rcnn.py	fix example that calls Tensor.__bool__ (#3650)	2024-03-07 16:59:26 -05:00
resnet.py	fp16 resnet (without expand backwards sum in float, doesn't work) (#3816)	2024-03-28 01:25:37 -04:00
retinanet.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
rnnt.py	create engine folder and move code (#3948)	2024-03-26 20:38:03 -07:00
transformer.py	replace with tensor op (#3099)	2024-01-12 14:13:40 -05:00
unet3d.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
vit.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00