pytorch注意事项

1.在pytorch中，创建好一个模型之后，需要将模型放入cuda才能调用GPU来进行训练。model.to(device)2.在

小女孩真可爱

193人浏览 · 2021-06-11 08:22:57

小女孩真可爱 · 2021-06-11 08:22:57 发布

1.在pytorch中，创建好一个模型之后，需要将模型放入cuda才能调用GPU来进行训练。

model.to(device)

2.在 pytorch中，使用nn.Conv2d()时，里面的padding的值是通过自己设置的。

$H_{out}=\frac{H_{in}+2P-dilation*(kernerl_size-1)-1}{strides} + 1$

这里的dilation是膨胀卷积，一般默认值是1.通过这个公式就可以反推出P的值了。

AI编程社区

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

AI编程工具大比拼：谁更胜一筹？你最喜欢那一款呢?

AI编程工具评测：精选4款主流产品对比分析当前AI编程工具呈现爆发式增长，本文重点评测了四款主流产品：Cursor（融合AI的智能IDE，适合复杂项目）、Augment（上下文理解超强的付费插件）、Trae（性价比高的国产编辑器）和CodeBuddy（腾讯出品的免费工具）。这些工具均支持最新AI模型，能实现代码生成、修改和解释功能，但在价格（20-50美元/月）、使用体验和功能侧重上各有差异。建