search

Sierkinhane / mtcnn-pytorch

A face detection algorithm
317 stars 162 forks source link

在训练rnet的时候loss会爆炸 有人遇到这个问题吗 #12

Open silvercherry opened 5 years ago

silvercherry commented 5 years ago

ValueError: could not broadcast input array from shape (350,933,3) into shape (350,0,3) 在用生成onet数据时 会报这个错

wanglaotou commented 4 years ago

我也遇到了，训练rnet，acc忽高忽低，loss值一直很大，第5个epoch的total loss能达到200多，一直怀疑是不是lr的问题，取的0.01，感觉太大了，虽然batchsize也取的很大。 请问下你现在已经解决了吗？