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