mnas训练 - Githubissues

ZHANG-SHI-CHANG / MnasNet-Tensorflow

MnasNet by Tensorflow and SNPE

15 stars 6 forks source link

Open alvinzhu123 opened 5 years ago

alvinzhu123 commented 5 years ago

你好，看完代码后，感觉是建了mnas的baseline网络结构，输出参数数目（不清楚有没有理解对）。没看懂代码中train和optimizer的作用，请问代码中有mnas基于各个block训练的吗？

ZHANG-SHI-CHANG commented 5 years ago

训练代码上传了 -_-，没有使用tensorflow的dataset接口，训练起来没有pytorch的快，方便的话可以使用我仓库里面pytorch的版本进行训练

alvinzhu123 commented 5 years ago

我在单片k80跑的train，没有用pytorch的那套代码，跑了三天还没有出best model。请问你当时训练的时候，用了多长时间？

ZHANG-SHI-CHANG commented 5 years ago

不好意思呀，我没看见。。我是在公司自己的数据集上训练的，18类场景识别，大约一天就可以收敛结束

baynaa7 commented 5 years ago

Hello, Have you ever trained the mnasnet with ssd?. I see through the source code and it seems, loss of detection just "pass"-ed.

MichaelGou1105 commented 5 years ago

您好，你使用过mnasnet做目标检测不？看你的代码里面，貌似一些检测的部分pass掉了