示例#1
0
def import_parameters():
    model = import_model()
    group_list = inception_v3.partition_model(model)
    group_list = [
        group for group in group_list if 'AuxLogits' not in group[0].fullname
    ]
    batch_list = [util.group_to_batch(group) for group in group_list]
    return batch_list
示例#2
0
def import_parameters():
    model = import_model()
    group_list = bert_base.partition_model(model)
    batch_list = [util.group_to_batch(group) for group in group_list]
    return batch_list