search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.21k stars 1.27k forks source link

训练loss问题 #644

Open Healer-EugeneLi opened 3 years ago

Healer-EugeneLi commented 3 years ago

你好，请问一下我使用了一些自己的结构，替换了回归分支跟分类分支的 self.conv_list = nn.ModuleList( [SeparableConvBlock(in_channels, in_channels, norm=False, activation=False) for i in range(num_layers)]) 里面的这个卷积模块，然后可以正常训练，一开始回归loss还在4跟3左右 但是一会之后回归loss直接不显示了，请问这是什么问题呢。分类loss是正常下降的。

zylo117 commented 3 years ago

不显示？康康？

Alan-D-Chen commented 3 years ago

您好，请问您在模型训练的时候（efficientdet-d2为例），大概用了多长时间？我这边是两个特斯拉V100 的GPU，使用efficientdet-d2模型作为预训练，head_only False，竟然需要240多个小时，但是查看服务器的GPU利用率却很低。这是怎么回事呢？ @zylo117