From c7b40dec1c453126f63dd2176e4dcd02a0280cb9 Mon Sep 17 00:00:00 2001 From: semihhdemirel Date: Tue, 18 Oct 2022 10:59:54 +0300 Subject: [PATCH] add import ImageDataset --- implementations/cyclegan/cyclegan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/implementations/cyclegan/cyclegan.py b/implementations/cyclegan/cyclegan.py index 445b5231..aae193b5 100644 --- a/implementations/cyclegan/cyclegan.py +++ b/implementations/cyclegan/cyclegan.py @@ -12,6 +12,7 @@ from torch.utils.data import DataLoader from torchvision import datasets from torch.autograd import Variable +from datasets import ImageDataset from models import * from datasets import *