- 博客(4)
- 收藏
- 关注
原创 使用梯度累积应对显存不足
梯度累积其实就是把一个大号的Batch拆成几个小的依次计算并累积梯度,然后进行一次参数更新。从而可以使用较小的显存来应对较大的BatchSize。
2024-01-09 19:38:14 389
原创 图像梯度损失的pytorch实现
关注图像纹理和边缘、支持Sobel、Prewitt、Roberts、Scharr这四种算子,默认 情况下使用Sobel算子,并且会对输入图像在各通道维度上计算平均值(也就是在默认情况下,对于输入的RGB图像,会首先计算得出其灰度图像再计算损失)
2023-12-15 18:38:43 767 1
原创 pytorch如何轻松获取特征图
那么如何知道模型中不同层的名称呢?一种方法是直接print打印模型,另一种方法是使用get_graph_node_names函数。通过torchvision提供的create_feature_extractor函数,我们可以根据层的名称轻松获取特征图。
2023-12-15 17:17:37 643
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人