---
datasets:
  - timm/mini-imagenet
metrics:
  - accuracy
tags:
  - Classification
  - arxiv:<2201.03545> # A ConvNet for the 2020s
pipeline_tag: image-classification
Operators:
  - Add: 12
  - Conv2D: 40
  - Flatten: 1
  - Gemme: 1
  - ReLu: 12


Source: PyTorch
Input:
  - size:
      - [N, 3, 300, 300]
  - format: NCHW
  - preprocessing:
      - Resize 
      - Normalise
Output:
  - size:
      - [N*100]

---

---
datasets:
  - timm/mini-imagenet
metrics:
  - accuracy
tags:
  - Classification
  - arxiv:<2201.03545> # A ConvNet for the 2020s
pipeline_tag: image-classification
Operators:
  - Add: 12
  - Conv2D: 40
  - Flatten: 1
  - Gemme: 1
  - ReLu: 12

features:
  - name: "ONNX import"
    tested_in_CI: true
  - name: "Export CPP"
    tested_in_CI: true


Source: PyTorch
Input:
  - size:
      - [N, 3, 300, 300]
  - format: NCHW
  - preprocessing:
      - Resize 
      - Normalise
Output:
  - size:
      - [N*100]

---