add dygraph fid score computation

1cd5b311 · lijianshe02 · ca9ff7e6 · 1cd5b311
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppgan/metric/test_fid_score.py ppgan/metric/test_fid_score.py +2 -2

未找到文件。
--- 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)