make pytorch-fid package optional

Plastickid · Dec 27, 2020 · f1356a3 · f1356a3
1 parent 5d12685
commit f1356a3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
@@ -29,7 +29,6 @@
 
 from tqdm import tqdm
 from einops import rearrange
-from pytorch_fid import fid_score
 
 from adabelief_pytorch import AdaBelief
 from gsa_pytorch import GSA
@@ -1139,6 +1138,7 @@ def evaluate(self, num = 0, num_image_tiles = 8, trunc = 1.0):
 
     @torch.no_grad()
     def calculate_fid(self, num_batches):
+        from pytorch_fid import fid_score
         torch.cuda.empty_cache()
 
         real_path = str(self.results_dir / self.name / 'fid_real') + '/'

diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py
@@ -1 +1 @@
-__version__ = '0.14.1'
+__version__ = '0.14.2'
diff --git a/setup.py b/setup.py
@@ -31,7 +31,6 @@
     'kornia',
     'numpy',
     'pillow',
-    'pytorch-fid',
     'retry',
     'scipy',
     'torch>=1.6',