add test image handler

2020-08-23 19:49:04 +08:00
parent 35ab7ecd51
commit 1e7f63cf85
3 changed files with 54 additions and 2 deletions
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -97,7 +97,7 @@ data:
            std: [0.5, 0.5, 0.5]
  test:
    dataloader:
-      batch_size: 4
+      batch_size: 8
      shuffle: False
      num_workers: 1
      pin_memory: False