duweidai / I2U-Net

11 stars 5 forks source link

Training problem #1

Open nunublbl opened 3 months ago

nunublbl commented 3 months ago

Dear Author: I use your model for polyp segmentation,but during the training, the loss don't change, which makes me confused.

ifyouarelihua commented 2 months ago

亲爱的作者:我使用您的模型进行息肉分割,但是在训练过程中,损失没有改变,这让我感到困惑。

Hi, I'm in the same situation, if you got it resolved can you tell me how?

nunublbl commented 2 months ago

the I2U-Net model's output use the sigmoid() finally. If your loss also apply sigmoid, you can delete the sigmoid() operation in the I2U-Net model's code.When I delete, it works.

------------------ 原始邮件 ------------------ 发件人: "duweidai/I2U-Net" @.>; 发送时间: 2024年6月27日(星期四) 中午11:05 @.>; @.**@.>; 主题: Re: [duweidai/I2U-Net] Training problem (Issue #1)

亲爱的作者:我使用您的模型进行息肉分割,但是在训练过程中,损失没有改变,这让我感到困惑。

Hi, I'm in the same situation, if you got it resolved can you tell me how?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

ifyouarelihua commented 2 months ago

I2U-Net 模型的输出最后使用 sigmoid()。如果您的损失也应用了 sigmoid,您可以在 I2U-Net 模型的代码中删除 sigmoid() 操作。当我删除时,它起作用了。 ------------------ 原始邮件 ------------------ 发件人: "duweidai/I2U-Net" @.>; 发送时间: 2024年6月27日(星期四) 中午11:05 @.>; @.**@.>; 主题: Re: [duweidai/I2U-Net] Training problem (Issue #1) 亲爱的作者:我使用您的模型进行息肉分割,但是在训练过程中,损失没有改变,这让我感到困惑。 Hi, I'm in the same situation, if you got it resolved can you tell me how? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

After I removed the sigmod() as you said, the loss values converged normally,thank you very much for your prompt reply. 666!

wenwen901 commented 2 months ago

I2U-Net 模型的输出最后使用 sigmoid()。如果您的损失也应用了 sigmoid,您可以在 I2U-Net 模型的代码中删除 sigmoid() 操作。当我删除时,它起作用了。 ------------------ 原始邮件 ------------------ 发件人: "duweidai/I2U-Net" @.**>; 发送时间: 2024年6月27日(星期四) 中午11:05 @.**>; @.**@.**>; 主题: Re: [duweidai/I2U-Net] Training problem (Issue #1) 亲爱的作者:我使用您的模型进行息肉分割,但是在训练过程中,损失没有改变,这让我感到困惑。 Hi, I'm in the same situation, if you got it resolved can you tell me how? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

After I removed the sigmod() as you said, the loss values converged normally,thank you very much for your prompt reply. 666!

请问一下,您在复现过程中使用的环境是多少呀,我看论文中没有给出这个项目所需的环境

ifyouarelihua commented 2 months ago

I2U-Net 模型的输出最后使用 sigmoid()。如果您的损失也应用了 sigmoid,您可以在 I2U-Net 模型的代码中删除 sigmoid() 操作。当我删除时,它起作用了。 ------------------ 原始邮件 ------------------ 发件人: "duweidai/I2U-Net" @.**>; 发送时间: 2024年6月27日(星期四) 中午11:05 @.**>; @.**@.**>; 主题: Re: [duweidai/I2U-Net] Training problem (Issue #1) 亲爱的作者:我使用您的模型进行息肉分割,但是在训练过程中,损失没有改变,这让我感到困惑。 Hi, I'm in the same situation, if you got it resolved can you tell me how? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

After I removed the sigmod() as you said, the loss values converged normally,thank you very much for your prompt reply. 666!

请问一下,您在复现过程中使用的环境是多少呀,我看论文中没有给出这个项目所需的环境

我复现的时候CUDA版本为12.2,torch为2.2.1,python=3.9 我在复现的时候并没有完整地将整个项目进行复现,只是选用了作者所提供的model部分,其余训练及测试框架及其超参数部分采用的是我自己的内容。

wenwen901 commented 2 months ago

I2U-Net 模型的输出最后使用 sigmoid()。如果您的损失也应用了 sigmoid,您可以在 I2U-Net 模型的代码中删除 sigmoid() 操作。当我删除时,它起作用了。 ------------------ 原始邮件 ------------------ 发件人: "duweidai/I2U-Net" @.**>; 发送时间: 2024年6月27日(星期四) 中午11:05 @.**>; @.**@.**>; 主题: Re: [duweidai/I2U-Net] Training problem (Issue #1) 亲爱的作者:我使用您的模型进行息肉分割,但是在训练过程中,损失没有改变,这让我感到困惑。 Hi, I'm in the same situation, if you got it resolved can you tell me how? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

After I removed the sigmod() as you said, the loss values converged normally,thank you very much for your prompt reply. 666!

请问一下,您在复现过程中使用的环境是多少呀,我看论文中没有给出这个项目所需的环境

我复现的时候CUDA版本为12.2,torch为2.2.1,python=3.9 我在复现的时候并没有完整地将整个项目进行复现,只是选用了作者所提供的model部分,其余训练及测试框架及其超参数部分采用的是我自己的内容。

好的感谢您

juingzhou commented 2 months ago

可以提供您的测试结果吗,我发现基于该代码的训练polyp数据集分割的结果与文中的结果差别有点大