search

pkuCactus / BDCN

The code for the CVPR2019 paper Bi-Directional Cascade Network for Perceptual Edge Detection
MIT License
341 stars 71 forks source link

训练速度很慢 #28

Open DreamerDoReMi opened 4 years ago

DreamerDoReMi commented 4 years ago

您好，我对您的论文很感兴趣， 现在在尝试跑您这份代码，但是loss.backword()的时候大概每张图需要0.6s, batch_size 设置为1时，每个iteration需要6s左右，我的显卡是1080Ti，系统是Ubuntu16.04. 请问您在训练的时候速度也是这么慢吗？

caihongzhiyue01 commented 4 years ago

您好，我对您的论文很感兴趣， 现在在尝试跑您这份代码，但是loss.backword()的时候大概每张图需要0.6s, batch_size 设置为1时，每个iteration需要6s左右，我的显卡是1080Ti，系统是Ubuntu16.04. 请问您在训练的时候速度也是这么慢吗？

您好，我想问下，下载的预训练模型里面有四个，按照我的理解，应该有一个就够了呢，为啥有这么多，难道不同数据集，不能叠加训练？

DreamerDoReMi commented 4 years ago

看一下参数文件的名字不就知道了

caihongzhiyue01 commented 4 years ago

看一下参数文件的名字不就知道了

我的意思是在BSDS500训练以后，如果要在NYUDv2或者Multicue进行训练，应该是重新训练吧，为何不能在原来训练的基础上进行训练？

singaloo commented 4 years ago

你好，我的训练速度也很慢，显卡跟您一样，您最后找到速度慢的原因了吗

DreamerDoReMi commented 4 years ago

没有，最近没看这个，但是我觉得你可以试试测测这个模型的计算量

---原始邮件--- 发件人: "singaloo"<notifications@github.com> 发送时间: 2019年12月8日(周日) 中午11:12 收件人: "pkuCactus/BDCN"<BDCN@noreply.github.com>; 抄送: "Author"<author@noreply.github.com>;"DreamerDoReMi"<365898376@qq.com>; 主题: Re: [pkuCactus/BDCN] 训练速度很慢 (#28)

你好，我的训练速度也很慢，显卡跟您一样，您最后找到速度慢的原因了吗

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

zcong17huang commented 3 years ago

你好，我的训练速度也很慢，显卡跟您一样，您最后找到速度慢的原因了吗

你好，我最近想跑通一下试试，结果发现训练特别慢，一个iteration需要好几秒，而且显存根本加载不上去，但是又确实在跑。很奇怪这种现象，不知道你是否也一样，或者你现在找到解决方法了嘛？