Loading zoo/wd14/sync.py +7 −3 Original line number Diff line number Diff line Loading @@ -4,7 +4,6 @@ from functools import lru_cache import numpy as np import onnx import onnxruntime import pandas as pd from ditk import logging from hbutils.string import plural_word Loading Loading @@ -47,7 +46,9 @@ _FC_NODE_PREFIXES_FOR_V3 = { "ViT_v3": ('core_model', 'head'), } if __name__ == '__main__': def sync(): import onnxruntime with TemporaryDirectory() as td: records = [] for model_name in tqdm(MODEL_NAMES): Loading @@ -59,7 +60,6 @@ if __name__ == '__main__': 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: Loading Loading @@ -151,3 +151,7 @@ if __name__ == '__main__': message=f'Upload {plural_word(len(df_records), "models")}', clear=True, ) if __name__ == '__main__': sync() Loading
zoo/wd14/sync.py +7 −3 Original line number Diff line number Diff line Loading @@ -4,7 +4,6 @@ from functools import lru_cache import numpy as np import onnx import onnxruntime import pandas as pd from ditk import logging from hbutils.string import plural_word Loading Loading @@ -47,7 +46,9 @@ _FC_NODE_PREFIXES_FOR_V3 = { "ViT_v3": ('core_model', 'head'), } if __name__ == '__main__': def sync(): import onnxruntime with TemporaryDirectory() as td: records = [] for model_name in tqdm(MODEL_NAMES): Loading @@ -59,7 +60,6 @@ if __name__ == '__main__': 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: Loading Loading @@ -151,3 +151,7 @@ if __name__ == '__main__': message=f'Upload {plural_word(len(df_records), "models")}', clear=True, ) if __name__ == '__main__': sync()