如何在pytorch中改变聚光灯下的默认优化，例如torch.optim.SGD？

作者: 哈哈哈哈
发布时间: 2024-12-18 07:39:09 (7天前)
转自：

2 条回复

0#
回复此人
一生流水 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你可以用 <code> partial </code> 从 <code> functools </code> 首先设置学习率和动力，然后通过这个课程 <code> ExplicitFactorizationModel </code> 。就像是： </p> <pre> <code> from functools import partial SDG_fix_lr_momentum = partial(torch.optim.SGD, lr=0.001, momentum=0.9) emodel = ExplicitFactorizationModel(n_iter=15, embedding_dim=32, use_cuda=False, loss='regression', l2=0.00005, optimizer_func=SDG_fix_lr_momentum) </code> </pre> </DIV>

编辑

登录后才能参与评论