diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py
index d578a798a32c7fc264989503b8b99b8b7da70f21..4fccd48da4d87152b1a71abdcdba71260901a9fa 100755
--- a/datasets/linemod/dataset.py
+++ b/datasets/linemod/dataset.py
@@ -25,6 +25,10 @@ import matplotlib.pyplot as plt
 class PoseDataset(data.Dataset):
     def __init__(self, mode, num, add_noise, root, noise_trans, refine):
         # ["banana1", "kiwi1", "pear2", "strawberry1", "apricot", "orange2", "peach1", "lemon2", "apple2" ]
+        #self.objlist = [1] #TODO
+
+        # apple, ....
+        #self.objlist = [1, 2, 3, 4, 5, 6, 7, 8]
         self.objlist = [1]
         self.mode = mode
 
@@ -91,6 +95,8 @@ class PoseDataset(data.Dataset):
         self.num_pt_mesh_large = 500
         self.num_pt_mesh_small = 500
         self.symmetry_obj_idx = [1] # TODO
+        #self.symmetry_obj_idx = [0 ,1, 3, 4, 5, 6]
+
 
     def __getitem__(self, index):
         img = Image.open(self.list_rgb[index])
@@ -107,16 +113,16 @@ class PoseDataset(data.Dataset):
         #             break
         # else:
 
-        print("---------------------------------------")
-        print("img : ", ori_img.shape)
+        #print("---------------------------------------")
+        #print("img : ", ori_img.shape)
         #print("obj",obj)
-        print("label : ", label.shape)
-        print("depth : ", depth.shape)
+        #print("label : ", label.shape)
+        #print("depth : ", depth.shape)
         
 
-        print("rank : ",rank)
-        print("len self.meta : ", len(self.meta))
-        print("len self.meta[obj] : ",len(self.meta[obj]))
+        #print("rank : ",rank)
+        #print("len self.meta : ", len(self.meta))
+        #print("len self.meta[obj] : ",len(self.meta[obj]))
         #print("keys : ", self.meta[obj].keys() )
         #print("key : ", self.meta[obj]['86156'])
         #print("self.meta[obj][rank] : ",self.meta[obj][rank] )
diff --git a/tools/train.py b/tools/train.py
index a2f29360b432360b90adeec9bae437c3728d2fec..61698f215b848136472dcc12da84f57af4af50a5 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -64,7 +64,7 @@ def main():
         opt.num_points = 500
         opt.outf = 'trained_models/linemod'
         opt.log_dir = 'experiments/logs/linemod'
-        opt.repeat_epoch = 10
+        opt.repeat_epoch = 20
     else:
         print('Unknown dataset')
         return