def import_task():
model = import_model()
func = import_func()
group_list = inception_v3.partition_model(model)
group_list = [
group for group in group_list if 'AuxLogits' not in group[0].fullname
]
shape_list = [util.group_to_shape(group) for group in group_list]
return model, func, shape_list
def import_task():
model = import_model()
func = import_func()
group_list = bert_base.partition_model(model)
shape_list = [util.group_to_shape(group) for group in group_list]
return model, func, shape_list