From b26241c965c13267d47b444bf4e9103d9424f973 Mon Sep 17 00:00:00 2001 From: wjtest1201 Date: Tue, 30 Aug 2022 17:48:03 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu/train.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gpu/train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/gpu/train.py b/gpu/train.py index ccedb05..c615864 100755 --- a/gpu/train.py +++ b/gpu/train.py @@ -64,8 +64,9 @@ if __name__ == '__main__': sgd.zero_grad() predict_y = model(train_x.float()) loss = cost(predict_y, train_label.long()) - if idx % 10 == 0: - print('idx: {}, loss: {}'.format(idx, loss.sum().item())) + #if idx % 10 == 0: + #print('idx: {}, loss: {}'.format(idx, loss.sum().item())) + print('idx: {}, loss: {}'.format(idx, loss.sum().item())) loss.backward() sgd.step()