diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/aligned_data_loader.py | 1 | ||||
| -rw-r--r-- | data/unaligned_data_loader.py | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/data/aligned_data_loader.py b/data/aligned_data_loader.py index bea3531..c5305fc 100644 --- a/data/aligned_data_loader.py +++ b/data/aligned_data_loader.py @@ -4,6 +4,7 @@ import torchvision.transforms as transforms from data.base_data_loader import BaseDataLoader from data.image_folder import ImageFolder from pdb import set_trace as st +# pip install future --upgrade from builtins import object class PairedData(object): diff --git a/data/unaligned_data_loader.py b/data/unaligned_data_loader.py index 4a06510..f37f6ce 100644 --- a/data/unaligned_data_loader.py +++ b/data/unaligned_data_loader.py @@ -2,6 +2,7 @@ import torch.utils.data import torchvision.transforms as transforms from data.base_data_loader import BaseDataLoader from data.image_folder import ImageFolder +# pip install future --upgrade from builtins import object from pdb import set_trace as st |