Copy Json to SaveDir in NLP Demo (#1252)

PaddlePaddle · Jul 5, 2022 · c590123 · c590123
1 parent 072ff7a
commit c590123
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/example/auto_compression/nlp/run.py b/example/auto_compression/nlp/run.py
@@ -4,6 +4,7 @@
 import functools
 from functools import partial
 import numpy as np
+import shutil
 import paddle
 import paddle.nn as nn
 from paddle.io import Dataset, BatchSampler, DataLoader
@@ -305,13 +306,17 @@ def main():
         if 'HyperParameterOptimization' not in all_config else eval_dataloader,
         eval_dataloader=eval_dataloader)
 
-    ac.compress()
+    if not os.path.exists(args.save_dir):
+        os.makedirs(args.save_dir)
+
     for file_name in os.listdir(global_config['model_dir']):
         if 'json' in file_name or 'txt' in file_name:
             shutil.copy(
                 os.path.join(global_config['model_dir'], file_name),
                 args.save_dir)
 
+    ac.compress()
+
 
 if __name__ == '__main__':
     paddle.enable_static()

diff --git a/example/auto_compression/pytorch_huggingface/run.py b/example/auto_compression/pytorch_huggingface/run.py
@@ -363,14 +363,17 @@ def main():
         'HyperParameterOptimization' not in all_config else eval_dataloader,
         eval_dataloader=eval_dataloader)
 
-    ac.compress()
+    if not os.path.exists(args.save_dir):
+        os.makedirs(args.save_dir)
 
     for file_name in os.listdir(global_config['model_dir']):
         if 'json' in file_name or 'txt' in file_name:
             shutil.copy(
                 os.path.join(global_config['model_dir'], file_name),
                 args.save_dir)
 
+    ac.compress()
+
 
 if __name__ == '__main__':
     paddle.enable_static()