diff --git a/gpu_new/train.py b/gpu_new/train.py
index 50da2a5..6912eb9 100755
--- a/gpu_new/train.py
+++ b/gpu_new/train.py
@@ -60,7 +60,7 @@ if __name__ == '__main__':
     epoch = args.epoch_size
     print('epoch_size is:{}'.format(epoch))
     for _epoch in range(epoch):
-        print('the {} epoch_size begin'.format(_epoch + 1))
+        #print('the {} epoch_size begin'.format(_epoch + 1))
         model.train()
         for idx, (train_x, train_label) in enumerate(train_loader):
             train_x = train_x.to(device)