Spaces:
Sleeping
Sleeping
Update finetune.py
Browse files- 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(
|