fix: GAN semantic visual output

jolibrain · Mar 26, 2024 · d3a5565 · d3a5565
1 parent 74b0e65
commit d3a5565
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -80,7 +80,7 @@ def launch_testing(opt, main_opt):
     with torch.no_grad():
         model.compute_metrics_test(dataloaders_test, epoch, total_iters)
 
-    metrics = model.get_current_metrics()
+    metrics = model.get_current_metrics([""])
     for metric, value in metrics.items():
         print(f"{metric}: {value}")