Spaces:
Running
on
A10G
Running
on
A10G
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(
|