未验证 提交 b03e5b8d 编写于 作者: L LielinJiang 提交者: GitHub

fix torch name (#385)

上级 099c595e
...@@ -26,7 +26,7 @@ from .builder import METRICS ...@@ -26,7 +26,7 @@ from .builder import METRICS
lpips = True 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' LINS_01_VGG_URL = 'https://paddlegan.bj.bcebos.com/models/lins_0.1_vgg.pdparams'
...@@ -94,7 +94,7 @@ class LPIPSMetric(paddle.metric.Metric): ...@@ -94,7 +94,7 @@ class LPIPSMetric(paddle.metric.Metric):
results_list = [] results_list = []
paddle.distributed.all_gather(results_list, results) paddle.distributed.all_gather(results_list, results)
self.results = paddle.concat(results_list).numpy() self.results = paddle.concat(results_list).numpy()
if len(self.results) <= 0: if len(self.results) <= 0:
return 0. return 0.
return np.mean(self.results) return np.mean(self.results)
......
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册