diff --git a/image_ref/main.py b/image_ref/main.py
index 9cd4c7b4cabfa940f5c1b31f59566a4fa49d8e79..d1b00ae7095e5c2fdde7e8c3ba079ccb16693523 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -19,6 +19,7 @@ def train_duo(model, data_train, optimizer, loss_function, epoch):
         param.requires_grad = True
 
     for imaer,imana, img_ref, label in data_train:
+        print(imaer.size())
         label = label.long()
         if torch.cuda.is_available():
             imaer = imaer.cuda()
@@ -46,6 +47,7 @@ def test_duo(model, data_test, loss_function, epoch):
         param.requires_grad = False
 
     for imaer,imana, img_ref, label in data_test:
+        print(imaer.size())
         label = label.long()
         if torch.cuda.is_available():
             imaer = imaer.cuda()