Update finetune.py

finetune.py

@@ -201,6 +201,12 @@ def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, wa
     log(tempdir_path, "exp/finetune/images")
     log(tempdir_path, "exp/finetune/config.yaml")
     log(tempdir_path, "exp/finetune/valid.acc.ave.pth")
+    log(tempdir_path, "output.log")
+    log(tempdir_path, "acc.png")
+    log(tempdir_path, "loss.png")
+    log(tempdir_path, "base.txt")
+    log(tempdir_path, "hyp.txt")
+    log(tempdir_path, "ref.txt")
 
     finetune_zip = zipfile.ZipFile(f"{tempdir_path}/finetune.zip", "w", zipfile.ZIP_DEFLATED)
     finetune_zip.write(f"{tempdir_path}/exp/stats")
@@ -208,6 +214,12 @@ def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, wa
     finetune_zip.write(f"{tempdir_path}/exp/finetune/images")
     finetune_zip.write(f"{tempdir_path}/exp/finetune/config.yaml")
     finetune_zip.write(f"{tempdir_path}/exp/finetune/valid.acc.ave.pth")
+    finetune_zip.write(f"{tempdir_path}/output.log")
+    finetune_zip.write(f"{tempdir_path}/acc.png")
+    finetune_zip.write(f"{tempdir_path}/loss.png")
+    finetune_zip.write(f"{tempdir_path}/base.txt")
+    finetune_zip.write(f"{tempdir_path}/hyp.txt")
+    finetune_zip.write(f"{tempdir_path}/ref.txt")
     finetune_zip.close()
 
     gr.Info("Finished generating result file in zip!")