diff --git a/norch/optim/optimizers/__pycache__/sgd.cpython-38.pyc b/norch/optim/optimizers/__pycache__/sgd.cpython-38.pyc index debbc02..c1b0789 100644 Binary files a/norch/optim/optimizers/__pycache__/sgd.cpython-38.pyc and b/norch/optim/optimizers/__pycache__/sgd.cpython-38.pyc differ 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)