From 8b0bd54ae9f8807c25eed552ec6917cae2095bf3 Mon Sep 17 00:00:00 2001
From: supershiye <38837128+supershiye@users.noreply.github.com>
Date: Tue, 29 Dec 2020 11:29:22 -0600
Subject: [PATCH] Update solver.py

The filename is no longer an argument in torchvision.
---
 solver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/solver.py b/solver.py
index 7d65766..30ab53d 100644
--- a/solver.py
+++ b/solver.py
@@ -415,7 +415,7 @@ def viz_traverse(self, limit=3, inter=2/3, loc=-1):
             for i, key in enumerate(Z.keys()):
                 for j, val in enumerate(interpolation):
                     save_image(tensor=gifs[i][j].cpu(),
-                               filename=os.path.join(output_dir, '{}_{}.jpg'.format(key, j)),
+                               fp=os.path.join(output_dir, '{}_{}.jpg'.format(key, j)),
                                nrow=self.z_dim, pad_value=1)
 
                 grid2gif(os.path.join(output_dir, key+'*.jpg'),