/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
python/torch_refs/torch_adam.py
22 строки
481 B
kolkir
Refactor python folder structure
01 мар 2025, 22:16
01 мар 2025, 22:16
ef25f05
Код
Авторство
О чём код?
import torch import torch.nn as nn params = [ torch.tensor([1, 2], dtype=torch.float32, requires_grad=True), torch.tensor([3, 4], dtype=torch.float32, requires_grad=True), ] optim = torch.optim.Adam(params) for i in range(4): for param in params: param.grad = torch.tensor([0.03, 0.02], dtype=torch.float32) optim.step() print("Adam params:") print(optim.state) print("updated weights:") for param in params: print(param.data)