Есть ли оптимизатор Levember-Marquardt или аналогичная альтернатива для Pytorch? - PullRequest
0 голосов
/ 29 января 2020

Я ищу оптимизатор LM или аналогичную оптимизацию второго порядка для pytorch. Для использования внутри модели NN. это код модели:

model = nn.Sequential(OrderedDict([ ('fc1', nn.Linear(D_in, H)),
                                    ('Tanh', nn.Tanh()),
                                    ('fc2', nn.Linear(H, D_out))]))
loss_fn = torch.nn.MSELoss(reduction='mean') ;
optimizer = torch.optim.AdamW(model.parameters(), lr=iLR, eps=iEPS,weight_decay=iwd)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...