fix torch name (PaddlePaddle#385)

AnnaTrainingG · Aug 6, 2021 · b03e5b8 · b03e5b8
1 parent 099c595
commit b03e5b8
Show file tree

Hide file tree

Showing 2 changed files with 270 additions and 105 deletions.
diff --git a/ppgan/metrics/lpips.py b/ppgan/metrics/lpips.py
@@ -26,7 +26,7 @@
 
 lpips = True
 
-VGG16_TORCHVISION_URL = 'https://paddlegan.bj.bcebos.com/models/vgg16_from_torch.pdparams'
+VGG16_TORCHVISION_URL = 'https://paddlegan.bj.bcebos.com/models/vgg16_official.pdparams'
 LINS_01_VGG_URL = 'https://paddlegan.bj.bcebos.com/models/lins_0.1_vgg.pdparams'
 
 
@@ -94,7 +94,7 @@ def accumulate(self):
             results_list = []
             paddle.distributed.all_gather(results_list, results)
             self.results = paddle.concat(results_list).numpy()
-            
+
         if len(self.results) <= 0:
             return 0.
         return np.mean(self.results)