ms180 commited on
Commit
57b1f30
1 Parent(s): a97446f

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +2 -0
finetune.py CHANGED
@@ -159,6 +159,8 @@ def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, wa
159
  finetune_config['scheduler_conf']['warmup_steps'] = warmup_steps
160
  finetune_config['multiple_iterator'] = False
161
  finetune_config['num_iters_per_epoch'] = None
 
 
162
 
163
  def build_model_fn(args):
164
  model, _ = S2TTask.build_model_from_file(
 
159
  finetune_config['scheduler_conf']['warmup_steps'] = warmup_steps
160
  finetune_config['multiple_iterator'] = False
161
  finetune_config['num_iters_per_epoch'] = None
162
+ finetune_config['multiprocessing_distributed'] = False
163
+ finetune_config['distributed'] = False
164
 
165
  def build_model_fn(args):
166
  model, _ = S2TTask.build_model_from_file(