Francis Lata eb2e59db42 RetinaNet model type annotations and loss functions (#9822) ... * add type annotations and loss functions for training * combine sum of multiple dims inside loss functions		2025-04-10 00:31:37 -04:00
..
bert.py	insert float() in bert acc (#9726)	2025-04-03 05:44:09 -04:00
clip.py	clip device fix (#6924)	2024-10-07 00:47:32 +08:00
convnext.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
efficientnet.py	remove the magic methods for moving between devices [pr] (#6881)	2024-10-04 20:27:52 +08:00
inception.py	Compute FID Score (#6802)	2024-10-01 19:47:58 -04:00
llama.py	olmoe (from stream, wip) (#9390)	2025-03-10 13:46:33 +08: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 model type annotations and loss functions (#9822)	2025-04-10 00:31:37 -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
unet.py	These casts should only happen if these are supported (#7644)	2024-11-12 07:56:50 +08: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