diff --git a/npu/pretrain_for_c2net.py b/npu/pretrain_for_c2net.py index a0be375..3b7318c 100755 --- a/npu/pretrain_for_c2net.py +++ b/npu/pretrain_for_c2net.py @@ -232,6 +232,9 @@ if __name__ == "__main__": if (args.epoch_size): epoch_size = args.epoch_size print('epoch_size is: ', epoch_size) + # 测试代码。结果回传 + os.system("cd /cache/script_for_grampus/ &&./uploader_for_npu " + "/cache/code/") + model.train(epoch_size, ds_train, callbacks=[time_cb, ckpoint_cb,