Skip to content

Commit

Permalink
dev(narugo): add wd14 syncer, ci skip
Browse files Browse the repository at this point in the history
  • Loading branch information
narugo1992 committed May 12, 2024
1 parent f8d873b commit 080716e
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions zoo/wd14/sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@

import numpy as np
import onnx
import onnxruntime
import pandas as pd
from ditk import logging
from hbutils.string import plural_word
Expand Down Expand Up @@ -47,7 +46,9 @@ def _seg_split(text):
"ViT_v3": ('core_model', 'head'),
}

if __name__ == '__main__':

def sync():
import onnxruntime
with TemporaryDirectory() as td:
records = []
for model_name in tqdm(MODEL_NAMES):
Expand All @@ -59,7 +60,6 @@ def _seg_split(text):
if model_name in _FC_NODE_PREFIXES_FOR_V3:
prefix = _FC_NODE_PREFIXES_FOR_V3[model_name]


def _is_fc(name):
return _seg_split(name)[:len(prefix)] == prefix
else:
Expand Down Expand Up @@ -151,3 +151,7 @@ def _is_fc(name):
message=f'Upload {plural_word(len(df_records), "models")}',
clear=True,
)


if __name__ == '__main__':
sync()

0 comments on commit 080716e

Please sign in to comment.