Commit 9504721f authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): try export onnx model for ccip

parent cc738c10

.github/workflows/export.yml

+14 −0
Original line number Diff line number Diff line
@@ -37,7 +37,21 @@ jobs: 
          pip install --upgrade pip

          pip install --upgrade flake8 setuptools wheel twine

          pip install -r requirements.txt

          pip install -r requirements-test.txt

          pip install -r requirements-zoo.txt

      - name: Prepare dataset

        uses: nick-fields/retry@v2

        env:

          CI: 'true'

          HF_NARUGO_USERNAME: ${{ secrets.HF_NARUGO_USERNAME }}

          HF_NARUGO_PASSWORD: ${{ secrets.HF_NARUGO_PASSWORD }}

        with:

          shell: bash

          timeout_minutes: 20

          max_attempts: 5

          retry_on: any

          command: |

            make dataset

      - name: Transform models to onnx format

        run: |

          python -m zoo.${{ matrix.model-name }} export -O ./${{ matrix.model-name }}

Makefile

+6 −1
Original line number Diff line number Diff line
@@ -55,3 +55,8 @@ dataset: 
	if [ ! -d ${DATASET_DIR}/monochrome_danbooru ]; then \

		git clone https://${HF_NARUGO_USERNAME}:${HF_NARUGO_PASSWORD}@huggingface.co/datasets/deepghs/monochrome_danbooru.git ${DATASET_DIR}/monochrome_danbooru; \

	fi

	if [ ! -d ${DATASET_DIR}/images_xtiny_v0 ]; then \

		mkdir -p ${DATASET_DIR}/images_xtiny_v0 && \

		wget -O ${DATASET_DIR}/images_xtiny_v0/images_xtiny_v0.tar.xz https://huggingface.co/datasets/deepghs/character_similarity/resolve/main/images_xtiny_v0.tar.xz && \

		cd ${DATASET_DIR}/images_xtiny_v0 && tar -xvf images_xtiny_v0.tar.xz && rm -rf *.tar.xz; \

	fi