fixed a file copying bug

justanhduc · Jul 17, 2020 · ae3cc3a · ae3cc3a
1 parent d079d47
commit ae3cc3a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/neuralnet_pytorch/monitor.py b/neuralnet_pytorch/monitor.py
@@ -15,7 +15,7 @@
 import logging
 from matplotlib import cm
 from imageio import imwrite
-import shutil
+from shutil import copyfile
 
 try:
     import visdom
@@ -866,7 +866,8 @@ def backup(self, files_or_folders):
         files_or_folders = (files_or_folders,) if isinstance(files_or_folders, str) else files_or_folders
         for f in files_or_folders:
             try:
-                utils.copytree(f, '%s/%s' % (self.file_folder, os.path.split(f)[-1]))
+                copy = copyfile if os.path.isfile(f) else utils.copytree
+                copy(f, '%s/%s' % (self.file_folder, os.path.split(f)[-1]))
             except FileNotFoundError:
                 root_logger.warning('No such file or directory: %s' % f)