Elias WahlandGitHub d2e3c391e8 Residual in MLM loss + Change default steps (#4935) ...

* Residual in mlm loss

* Reduce default steps to 160K * 24

* oops

* comment

2024-06-12 16:09:18 -04:00

..

bert.py

Residual in MLM loss + Change default steps (#4935)

2024-06-12 16:09:18 -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

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04:00

mask_rcnn.py

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04: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

change Tensor.stack to method (#4719)

2024-05-24 17:04:19 -04: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