From 9634ab12c5e158bcdbe4a913c6de0e91b60f6db4 Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Tue, 7 May 2024 01:27:48 -0300 Subject: [PATCH] Fix long time step() detach --- .../optimizers/__pycache__/sgd.cpython-38.pyc | Bin 1239 -> 1239 bytes test.py | 5 ++--- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/norch/optim/optimizers/__pycache__/sgd.cpython-38.pyc b/norch/optim/optimizers/__pycache__/sgd.cpython-38.pyc index debbc0286adc72080bc37acaac9cc825c00b7d24..c1b0789c9b83c4adc74041b9646e684dac65d982 100644 GIT binary patch delta 19 Zcmcc4d7YChl$V!_0SKhmZsa=40st=a1egE- delta 19 Zcmcc4d7YChl$V!_0SH=FZ{#}50st{z1pfd4 diff --git a/test.py b/test.py index 0fc7607..3fc72da 100644 --- a/test.py +++ b/test.py @@ -97,7 +97,7 @@ if __name__ == "__main__": input_list = [[0.5 for _ in range(100)]] input = norch.Tensor(input_list).T criterion = nn.MSELoss() - optimizer = norch.optim.SGD(modelo.parameters(), lr=0.05) + optimizer = norch.optim.SGD(modelo.parameters(), lr=0.1) target_list = [[random.random() for _ in range(2)]] target = norch.Tensor(target_list).T @@ -111,9 +111,8 @@ if __name__ == "__main__": optimizer.zero_grad() loss.backward() - #print('fora grad', modelo.layer1.weight.grad, "\n\n") optimizer.step() - print(loss) + #print(loss) fim = time.time() print(fim - ini)