Merge branch 'zuoym15-master'

lanpa · Oct 20, 2019 · c9ecc54 · c9ecc54
2 parents 8dcac01 + 377b5f9
commit c9ecc54
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tensorboardX/summary.py b/tensorboardX/summary.py
@@ -355,9 +355,12 @@ def make_video(tensor, fps):
     clip = mpy.ImageSequenceClip(list(tensor), fps=fps)
 
     filename = tempfile.NamedTemporaryFile(suffix='.gif', delete=False).name
-    try:  # older version of moviepy does not support progress_bar argument.
-        clip.write_gif(filename, verbose=False, progress_bar=False)
+
+    # moviepy >= 1.0.0 use logger=None to suppress output.
+    try:
+        clip.write_gif(filename, verbose=False, logger=None)
     except TypeError:
+        logging.warning('Upgrade to moviepy >= 1.0.0 to supress the progress bar.')
         clip.write_gif(filename, verbose=False)
 
     with open(filename, 'rb') as f: