Browse Source

非下沉模式

master
liupengfei 1 month ago
parent
commit
b759027d71
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train.py

+ 1
- 1
train.py View File

@@ -202,7 +202,7 @@ def run_train():
ckpt_save_dir = set_save_ckpt_dir()
add_ckpt_callback(step_size, ckpt_save_dir, cbs)
add_eval_callback(model, ckpt_save_dir, cbs)
model.train(config.epoch_size, train_dataset, callbacks=cbs, dataset_sink_mode=True)
model.train(config.epoch_size, train_dataset, callbacks=cbs, dataset_sink_mode=False)
# profiles.analyse()

def _get_rank_info():


Loading…
Cancel
Save