mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-09 23:12:08 +08:00
dd778596b7
date: 2025-03-15T21:10:51 master commit: fb7b9c0f9420d228f03362970ebcfb7237095cf3
20 lines
533 B
Plaintext
20 lines
533 B
Plaintext
Each model should be a clean single file.
|
|
They are imported from the top level `models` directory
|
|
|
|
It should be capable of loading weights from the reference imp.
|
|
|
|
We will focus on these 5 models:
|
|
|
|
# Resnet50-v1.5 (classic) -- 8.2 GOPS/input
|
|
# Retinanet
|
|
# 3D UNET (upconvs)
|
|
# RNNT
|
|
# BERT-large (transformer)
|
|
|
|
They are used in both the training and inference benchmark:
|
|
https://mlcommons.org/en/training-normal-21/
|
|
https://mlcommons.org/en/inference-edge-30/
|
|
And we will submit to both.
|
|
|
|
NOTE: we are Edge since we don't have ECC RAM
|