From 1cd5b31128c641f4915368d63a707374acf1b996 Mon Sep 17 00:00:00 2001 From: lijianshe02 Date: Thu, 20 Aug 2020 06:16:05 +0000 Subject: [PATCH] add dygraph fid score computation --- ppgan/metric/test_fid_score.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppgan/metric/test_fid_score.py b/ppgan/metric/test_fid_score.py index 6e484f5..e8abcca 100644 --- a/ppgan/metric/test_fid_score.py +++ b/ppgan/metric/test_fid_score.py @@ -51,8 +51,8 @@ def main(): batch_size = args.batch_size with fluid.dygraph.guard(): - fid_value = calculate_fid_given_paths(paths, inference_model_path, 1, - True, 2048) + fid_value = calculate_fid_given_paths(paths, inference_model_path, + batch_size, args.use_gpu, 2048) print('FID: ', fid_value) -- GitLab