hooved c2689c505e Clip model updates for Stable Diffusion mlperf training (#12313) ...

* stable diffusion mlperf clip changes

* add clip tests

* set gelu as attribute

* add more tests

* factor out GPUS

* rerun CI

* add imports to if blocks

* remove unneeded axis

* add clip tests to CI

* move clip tests

* add deps, disable max buf size

2025-09-29 21:50:14 -04:00

..

bert.py

insert float() in bert acc (#9726)

2025-04-03 05:44:09 -04:00

clip.py

Clip model updates for Stable Diffusion mlperf training (#12313)

2025-09-29 21:50:14 -04:00

convnext.py

remove Tensor.no_grad, it's meaningless now [pr] (#10556)

2025-05-28 22:20:02 -07:00

efficientnet.py

remove the magic methods for moving between devices [pr] (#6881)

2024-10-04 20:27:52 +08:00

inception.py

don't hardcode weights path (#12171)

2025-09-15 00:33:47 -04:00

llama.py

remove llvm_bf16_cast (#12075)

2025-09-08 20:51:15 +03:00

mask_rcnn.py

move BoxCoder to mlperf helpers (#9773)

2025-04-07 20:27:06 -04:00

resnet.py

Fix FC layer ResNet load_from_pretrained error (#8387)

2024-12-26 18:11:27 -05:00

retinanet.py

RetinaNet INITMLPERF support (#9950)

2025-04-21 10:32:05 -04:00

rnnt.py

change Tensor.stack to method (#4719)

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

t5.py

Flux.1 (#6334)

2024-09-24 10:08:04 +08:00

transformer.py

_one_hot_along_dim input needs to be int (#9179)

2025-02-20 09:00:43 -05:00

unet3d.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

unet.py

These casts should only happen if these are supported (#7644)

2024-11-12 07:56:50 +08:00

vit.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00