polish download pretrain
parent
6675cf4312
commit
e3ca763dee
|
@ -16,8 +16,6 @@ import architectures
|
||||||
import types
|
import types
|
||||||
from difflib import SequenceMatcher
|
from difflib import SequenceMatcher
|
||||||
|
|
||||||
from ppcls.utils.config import get_config
|
|
||||||
|
|
||||||
|
|
||||||
def get_architectures():
|
def get_architectures():
|
||||||
"""
|
"""
|
||||||
|
@ -30,13 +28,6 @@ def get_architectures():
|
||||||
return names
|
return names
|
||||||
|
|
||||||
|
|
||||||
def get_pretrained():
|
|
||||||
"""
|
|
||||||
get all of model pretrained
|
|
||||||
"""
|
|
||||||
return get_config('./pretrained.list')
|
|
||||||
|
|
||||||
|
|
||||||
def similar_architectures(name='', names=[], thresh=0.1, topk=10):
|
def similar_architectures(name='', names=[], thresh=0.1, topk=10):
|
||||||
"""
|
"""
|
||||||
inferred similar architectures
|
inferred similar architectures
|
||||||
|
|
|
@ -20,7 +20,6 @@ import sys
|
||||||
|
|
||||||
import paddle.fluid as fluid
|
import paddle.fluid as fluid
|
||||||
|
|
||||||
from ppcls.modeling import get_architectures
|
|
||||||
from ppcls.modeling import similar_architectures
|
from ppcls.modeling import similar_architectures
|
||||||
from ppcls.utils import logger
|
from ppcls.utils import logger
|
||||||
|
|
||||||
|
|
|
@ -23,9 +23,9 @@ import tarfile
|
||||||
import tqdm
|
import tqdm
|
||||||
import zipfile
|
import zipfile
|
||||||
|
|
||||||
from ppcls.modeling import get_pretrained
|
|
||||||
from ppcls.modeling import similar_architectures
|
from ppcls.modeling import similar_architectures
|
||||||
from ppcls.utils.check import check_architecture
|
from ppcls.utils.check import check_architecture
|
||||||
|
from ppcls.utils.config import get_config
|
||||||
from ppcls.utils import logger
|
from ppcls.utils import logger
|
||||||
|
|
||||||
__all__ = ['get']
|
__all__ = ['get']
|
||||||
|
@ -172,7 +172,7 @@ def _decompress(fname):
|
||||||
def _check_pretrained_name(architecture):
|
def _check_pretrained_name(architecture):
|
||||||
assert isinstance(architecture, str), \
|
assert isinstance(architecture, str), \
|
||||||
("the type of architecture({}) should be str". format(architecture))
|
("the type of architecture({}) should be str". format(architecture))
|
||||||
similar_names = similar_architectures(architecture, get_pretrained())
|
similar_names = similar_architectures(architecture, get_config('../../../configs/pretrained.list'))
|
||||||
model_list = ', '.join(similar_names)
|
model_list = ', '.join(similar_names)
|
||||||
err = "{} is not exist! Maybe you want: [{}]" \
|
err = "{} is not exist! Maybe you want: [{}]" \
|
||||||
"".format(architecture, model_list)
|
"".format(architecture, model_list)
|
||||||
|
|
Loading…
Reference in New Issue