Browse Source

更新 'npu/pretrain_for_c2net.py'

test_v20221116
wjtest1215 2 years ago
parent
commit
a7847dd1e3
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      npu/pretrain_for_c2net.py

+ 3
- 0
npu/pretrain_for_c2net.py View File

@@ -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,


Loading…
Cancel
Save