Skip to content

Commit 687943d

Browse files
committed
update agent workflow
1 parent 811933b commit 687943d

2 files changed

Lines changed: 16 additions & 8 deletions

File tree

.agent/workflows/port-new-model.md

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,11 +106,19 @@ Key rules:
106106

107107
## 6. Documentation
108108

109-
1. **Model-specific `README.md`** (e.g., `beit/README.md`, `llama2/README.md`):
109+
Model table metrics:
110+
- **Params**: Computed via `model_surgery.count_params(model)`.
111+
- **FLOPs**: Computed via `model_surgery.get_flops(model)`.
112+
- **Top1 Acc**: From the original paper or reference repo.
113+
114+
1. **Inner README** (e.g., `beit/README.md`, `llama2/README.md`):
110115
- Add paper reference in the Summary section.
111-
- Add model table with Params, FLOPs, and model-specific metrics.
112-
- Download links: `[filename.h5](https://github.com/leondgarse/keras_cv_attention_models/releases/download/{sub_release}/{filename}.h5)`
113-
2. **Root `README.md`**: Add entry to Table of Contents in alphabetical order.
116+
- Add model table with Params, FLOPs, Input, Top1 Acc, and a separate **Download** column with links.
117+
- Example: `| ViT5_Small_Patch16 | 22.04M | 4.73G | 224 | 82.2 | [vit5_small_patch16_224.h5](https://...h5) |`
118+
2. **Outer `README.md`**:
119+
- Add entry to Table of Contents in alphabetical order.
120+
- Add model table where the **model name itself is the download link** (no separate Download column).
121+
- Example: `| [ViT5_Small_Patch16](https://...h5) | 22.04M | 4.73G | 224 | 82.2 |`
114122
3. **`__init__.py`**: Ensure all parameters are documented with model-specific defaults noted.
115123

116124
## 7. Testing

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1527,10 +1527,10 @@
15271527

15281528
| Model | Params | FLOPs | Input | Top1 Acc |
15291529
| ------------------ | ------ | ------ | ----- | -------- |
1530-
| ViT5_Small_Patch16 | 22.04M | 4.73G | 224 | 82.2 |
1531-
| ViT5_Base_Patch16 | 86.54M | 18.00G | 224 | 84.2 |
1532-
| ViT5_Base_Patch16 | 86.83M | 56.19G | 384 | 85.4 |
1533-
| ViT5_Large_Patch16 | 304.3M | 63.01G | 224 | 84.9 |
1530+
| [ViT5_Small_Patch16](https://github.com/leondgarse/keras_cv_attention_models/releases/download/beit/vit5_small_patch16_224_imagenet.h5) | 22.04M | 4.73G | 224 | 82.2 |
1531+
| [ViT5_Base_Patch16](https://github.com/leondgarse/keras_cv_attention_models/releases/download/beit/vit5_base_patch16_224_imagenet.h5) | 86.54M | 18.00G | 224 | 84.2 |
1532+
| [ViT5_Base_Patch16](https://github.com/leondgarse/keras_cv_attention_models/releases/download/beit/vit5_base_patch16_384_imagenet.h5) | 86.83M | 56.19G | 384 | 85.4 |
1533+
| [ViT5_Large_Patch16](https://github.com/leondgarse/keras_cv_attention_models/releases/download/beit/vit5_large_patch16_224_imagenet.h5) | 304.3M | 63.01G | 224 | 84.9 |
15341534
| ViT5_Large_Patch16 | 304.6M | 193.2G | 384 | 86.0 |
15351535
## VOLO
15361536
- [Keras VOLO](keras_cv_attention_models/volo) is for [PDF 2106.13112 VOLO: Vision Outlooker for Visual Recognition](https://arxiv.org/pdf/2106.13112.pdf).

0 commit comments

Comments
 (0)