fix small bug in U-GAT-IT

2020-09-11 22:34:43 +08:00
parent 72d09aa483
commit 85b5c3f589
2 changed files with 3 additions and 2 deletions
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -92,6 +92,7 @@ data:
            mean: [ 0.5, 0.5, 0.5 ]
            std: [ 0.5, 0.5, 0.5 ]
  test:
+    which: video_dataset
    dataloader:
      batch_size: 8
      shuffle: False