forked from huawei/mindspore2022
add print
This commit is contained in:
parent
0e7844daa9
commit
ad8ecbc3af
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue