def _check_pretrained_name(architecture): assert isinstance(architecture, str), \ ("the type of architecture({}) should be str". format(architecture)) pretrained = _get_pretrained() similar_names = similar_architectures(architecture, pretrained) model_list = ', '.join(similar_names) err = "{} is not exist! Maybe you want: [{}]" \ "".format(architecture, model_list) if architecture not in similar_names: raise ModelNameError(err)
def _check_pretrained_name(architecture): assert isinstance(architecture, str), \ ("the type of architecture({}) should be str". format(architecture)) with open('./configs/pretrained.list') as flist: pretrained = [line.strip() for line in flist] similar_names = similar_architectures(architecture, pretrained) model_list = ', '.join(similar_names) err = "{} is not exist! Maybe you want: [{}]" \ "".format(architecture, model_list) if architecture not in similar_names: raise ModelNameError(err)
def check_architecture(architecture): """ check architecture and recommend similar architectures """ assert isinstance(architecture, str), \ ("the type of architecture({}) should be str". format(architecture)) similar_names = similar_architectures(architecture) model_list = ', '.join(similar_names) err = "{} is not exist! Maybe you want: [{}]" \ "".format(architecture, model_list) try: assert architecture in similar_names except AssertionError: logger.error(err) sys.exit(1)
def check_architecture(architecture): """ check architecture and recommend similar architectures """ assert isinstance(architecture, dict), \ ("the type of architecture({}) should be dict". format(architecture)) assert "name" in architecture, \ ("name must be in the architecture keys, just contains: {}". format(architecture.keys())) similar_names = similar_architectures(architecture["name"], get_architectures()) model_list = ', '.join(similar_names) err = "{} is not exist! Maybe you want: [{}]" \ "".format(architecture["name"], model_list) try: assert architecture["name"] in similar_names except AssertionError: logger.error(err) sys.exit(1)