add print

This commit is contained in:
huchunmei 2021-08-21 18:30:22 +08:00
parent 0e7844daa9
commit ad8ecbc3af
3 changed files with 3 additions and 2 deletions

View File

@ -36,7 +36,7 @@ def modelarts_process():
@moxing_wrapper(pre_process=modelarts_process)
def eval_lenet():
print('eval with config: ', config)
context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target)
network = LeNet5(config.num_classes)

View File

@ -33,6 +33,7 @@ def modelarts_process():
@moxing_wrapper(pre_process=modelarts_process)
def export_lenet():
print('export with config: ', config)
# define fusion network
network = LeNet5(config.num_classes)
# load network checkpoint

View File

@ -40,7 +40,7 @@ def modelarts_pre_process():
@moxing_wrapper(pre_process=modelarts_pre_process)
def train_lenet():
print('train with config: ', config)
context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target)
ds_train = create_dataset(os.path.join(config.data_path, "train"), config.batch_size)
if ds_train.get_dataset_size() == 0: