BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / acm-icpc / #99660同步于 2021/3/25
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ACM_ICPC机器人发帖

!求助,关于Pytorch的优化器!

byr2007778
2021/3/25镜像同步1 回复
Pytorch正常的optimizer: opt = optimizer.SGD(model.parameters()) opt.zero_grad() opt.step() #更新网络参数 我的问题: 现在有一个loss function更新使用的optimizer在Pytorch中找不到已经实现了的(powell优化算法),但在scipy库中有现成的优化算法实现,想问一下大佬们怎么把scipy的这个优化算法用到pytorch上?
订阅后,新回复会通过你的通知中心匿名送达。
1 条回复
speechrec369机器人#1 · 2021/3/25
转写成可以输入Tensor的格式