Update finetune.py

finetune.py

@@ -160,6 +160,7 @@ def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, wa
     finetune_config['scheduler_conf']['warmup_steps'] = warmup_steps
     finetune_config['multiple_iterator'] = False
     finetune_config['num_iters_per_epoch'] = None
+    finetune_config['multiprocessing_distributed'] = False
 
     def build_model_fn(args):
         model, _ = S2TTask.build_model_from_file(
@@ -198,15 +199,13 @@ def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, wa
     log(tempdir_path, "exp/s2t_stats_raw_bpe50000")
     log(tempdir_path, "exp/finetune/tensorboard")
     log(tempdir_path, "exp/finetune/images")
-    log(tempdir_path, "exp/finetune/train.log")
     log(tempdir_path, "exp/finetune/config.yaml")
     log(tempdir_path, "exp/finetune/valid.acc.ave.pth")
 
     finetune_zip = zipfile.ZipFile(f"{tempdir_path}/finetune.zip", "w", zipfile.ZIP_DEFLATED)
-    finetune_zip.write(f"exp/s2t_stats_raw_bpe50000")
+    finetune_zip.write(f"{tempdir_path}/exp/stats")
     finetune_zip.write(f"{tempdir_path}/exp/finetune/tensorboard")
     finetune_zip.write(f"{tempdir_path}/exp/finetune/images")
-    finetune_zip.write(f"{tempdir_path}/exp/finetune/train.log")
     finetune_zip.write(f"{tempdir_path}/exp/finetune/config.yaml")
     finetune_zip.write(f"{tempdir_path}/exp/finetune/valid.acc.ave.pth")
     finetune_zip.close()