torch框架
文章平均质量分 55
知福致福
这个作者很懒,什么都没留下…
展开
-
将神经网络加到列表里面之后,用.to(device)函数迁移参数到gpu上失败
使用[nn.Linear(10,10) for i in range(n)]生成多个相同的神经网络被重复调用,独自训练参数,结果参数迁移的时候发现只有列表的第一个对象的参数被迁移到gpu上了,其他参数都没有被迁移过去。nn.ModuleList 这个类,可以把任意 nn.Module 的子类 (比如 nn.Conv2d, nn.Linear 之类的) 加到这个 list 里面,类似于Python 自带的 list。如果只使用python自带的list,会发现。原创 2024-04-13 20:02:06 · 386 阅读 · 0 评论 -
梯度反向传播过程是如何处理repeat函数的
梯度反向传播原创 2024-04-04 15:06:28 · 436 阅读 · 0 评论 -
搭建神经网络(torch.nn的用法)
零零碎碎总结了一些torch框架里面nn模块的用法,尤其是关于搭建神经网络的。原创 2023-11-09 16:23:05 · 1123 阅读 · 0 评论