diff --git a/gpu/train_fail3.py b/gpu/train_fail3.py index 5854d07..1bc70bb 100644 --- a/gpu/train_fail3.py +++ b/gpu/train_fail3.py @@ -69,8 +69,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(gettime(), 'idx: {}, loss: {}'.format(idx, loss.sum().item())) + #if idx % 10 == 0: + #print(gettime(), 'idx: {}, loss: {}'.format(idx, loss.sum().item())) + print(gettime(), 'idx: {}, loss: {}'.format(idx, loss.sum().item())) loss.backward() sgd.step()