Alex Cheema
e1cb840b31
Merge commit '0870e6bfb0d46d58e29c6d5822e8bf629d03bdcc' as 'tinychat'
|
1 سال پیش | |
|---|---|---|
| .. | ||
| training_submission_v4.0 | 1 سال پیش | |
| README | 1 سال پیش | |
| dataloader.py | 1 سال پیش | |
| helpers.py | 1 سال پیش | |
| initializers.py | 1 سال پیش | |
| losses.py | 1 سال پیش | |
| lr_schedulers.py | 1 سال پیش | |
| metrics.py | 1 سال پیش | |
| model_eval.py | 1 سال پیش | |
| model_spec.py | 1 سال پیش | |
| model_train.py | 1 سال پیش | |
README
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
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