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