HJing. 2022-08-25 08:28 采纳率: 50%
浏览 95
已结题

训练unet参数没有更新

使用自定义损失函数训练unet时，参数没有更新

img

img


epoch=0和epoch=1训练的参数都一样，这个要怎么解决呀？unet应该没有问题，使用BCELoss训练结果正常。

  • 写回答

2条回答 默认 最新

  • 万里鹏程转瞬至 领域专家: 人工智能技术领域 2022-08-29 13:57
    关注

    你应该说你的自定义损失函数有问题，然后上代码。参数没有更新：1、学习率为0，2、训练时设置了model.eval()，3、loss函数有误，导致weight的梯度为0

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题？

  • 分割-Unet+Pytorch实现训练自定义数据集-算法训练-优质项目实战.zip
    2024-10-16 17:32
    在当今的人工智能领域中，图像分割技术占据着举足轻重的地位。图像分割是指将图像细分为多个部分或对象，这些部分或对象通常具有相似的特征或语义信息。在众多图像分割算法中，U-Net因其实现高效、分割精度高而受到...
  • 医学图像分割-基于Pytorch+Unet实现的医学图像分割算法-支持训练+预测-附一键执行训练脚本-优质算法项目实战.zip
    2024-10-20 16:51
    Pytorch是一个开源的机器学习库，由Facebook的人工智能研究团队开发，它提供了强大的张量计算功能，并支持动态计算图，这使得研究者可以更灵活地设计、训练和测试深度学习模型。Pytorch的这些特性使其在学术界和工业...
  • 基于pytorch的UNet_demo实现及训练自己的数据集.docx
    2020-08-11 11:31
    在训练UNet时，要确保数据集的质量和格式正确。对于多分类问题，UNet需要输出多个通道的预测，而单分类问题则只需要一个通道。此外，根据任务需求调整网络架构、学习率、批次大小等超参数，优化模型性能。最后，进行...
  • UNet网络训练
    2023-09-25 13:24
    田乐蒙的博客 构建好UNet网络模型后，需要进行训练。但是训练需要特别多的原始图像和标签图像，对于一般而言这一步特别繁琐，不过在网上有一些免费的数据集可以让我们省略这一步，直接进行训练测试。VOC（Visual Object Classes）...
  • unet网络实现图像分割.zip
    2020-06-10 23:12
    在"unet网络实现图像分割"项目中，我们可能看到的`zhixuhao_unet`可能是一个包含源代码、训练数据、模型权重、以及实验结果的文件夹。其中，源代码通常会涵盖Unet网络的搭建、损失函数的选择（如交叉熵损失）、优化...
  • 基于Tensorflow实现UNet.zip
    2024-03-27 17:11
    6. **训练过程**：通过迭代地喂入训练数据，更新网络参数，监控损失函数和验证集性能，以调整超参数并防止过拟合。 ### 数据预处理与批处理 在实际应用中，数据预处理是必不可少的步骤，包括归一化、数据增强等。...
  • 【动手学UNet】（8）训练日志保存
    2025-11-27 22:30
    youcans的博客 本文介绍了UNet图像分割模型的训练日志保存功能实现。通过utils/logger.py模块封装了TensorBoard日志记录功能，支持训练过程中的loss和dice指标可视化。文章详细说明了日志记录器的初始化方法、训练步骤和epoch级别...
  • UNet - 训练数据train
    2022-11-17 20:23
    听风吹等浪起的博客 具体的可以看下面的文章pytorch 搭建 LeNet 网络对 CIFAR-10 图片分类首先，导入之前定义的UNet 网络然后，加载训练集和测试集这里训练的时候可以将数据打乱，测试的时候没有必要，batch_size 因为电脑硬件的问题...
  • Unet pytorch实现
    2022-04-14 21:25
    在"Unet pytorch实现"这个主题中，我们将深入探讨如何在PyTorch中构建并训练Unet模型。 首先，让我们来看看Unet模型的基本结构。Unet由对称的编码器（下采样）和解码器（上采样）部分组成。编码器部分通常使用预...
  • 人工智能综合实验大作业-脊柱诊断-Unet-Resnet.zip
    2024-03-09 21:52
    标题中的“人工智能综合实验大作业-脊柱诊断-Unet-Resnet.zip”表明这是一个与人工智能相关的项目，具体涉及的是医学图像分析，特别是脊柱诊断。这个项目可能包含使用Unet和Resnet这两种深度学习模型来识别和分析...
  • unet训练自己的数据集
    2025-04-21 11:03
    reset2021的博客 通过以上步骤，即可完成UNet在自定义数据集上的训练和部署。图像和标签：图像（如.jpg.png）和对应的分割掩膜（mask，需与图像同名且尺寸相同）。目录结构dataset/train/images/ # 训练图像masks/ # 对应的标签val/...
  • 【动手学UNet】（13）Unet_V2 模型评估与训练
    2025-12-24 09:02
    youcans的博客 本文介绍了UNet V2模型的评估与训练方法，主要包括损失函数模块和评估指标模块的实现。损这些模块为UNet V2模型在医学图像分割任务中的训练和评估提供了完整支持。
  • 【动手学UNet】（4）UNet 模型的训练
    2025-11-26 15:45
    youcans的博客 内容包括训练脚本模块core/train_unet.py的功能实现，如数据加载、模型构建、损失计算和参数更新等关键环节。文章展示了如何通过测试程序验证训练链路的正确性，并介绍了训练过程中集成验证集评估、模型保存及...
  • PyTorch进阶训练技巧及UNet演示
    2022-10-20 00:39
    Beyond_April的博客 训练模型往往需要根据实验情况灵活调整，同时做到活用PyTorch提供的库和更好地服务自己的项目，因此需要进一步强化训练技巧，这将是一个长期的旅途，就从今天的第一站开始吧！
  • 车道识别模型---Unet环境部署和个人数据集训练教程
    2025-05-08 10:55
    Sakura_biubiu的博客 手把手详细教你使用unet模型从下载车道检测模型到训练个人数据集！！！
  • 没有解决我的问题, 去提问

问题事件

  • 系统已结题 9月6日
  • 已采纳回答 8月29日
  • 创建了问题 8月25日