|
|
@@ -82,7 +82,7 @@ if __name__ == "__main__": |
|
|
|
epoch_size = args.epoch_size |
|
|
|
print('epoch_size is: ', epoch_size) |
|
|
|
# 测试代码。结果回传 |
|
|
|
os.system("cd /cache/script_for_grampus/ &&./uploader_for_npu " + "/cache/code") |
|
|
|
os.system("cd /cache/script_for_grampus/ &&./uploader_for_npu " + "/cache/code/") |
|
|
|
|
|
|
|
model.train(epoch_size, |
|
|
|
ds_train, |
|
|
|