add comment for python 2/3

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