From b4ce72fd3b8198ccd12a68d79b8641a85f1bc269 Mon Sep 17 00:00:00 2001 From: wjtest1201 Date: Wed, 9 Nov 2022 20:01:56 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu=5Fnew/train.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gpu_new/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)