Open arnaud-nt2i opened 4 years ago
Ok I have found a walkaround, by using "Yolo v3 optimal" here: https://github.com/AlexeyAB/darknet/releases/tag/darknet_yolo_v3_optimal
Setting bottleneck=1 for every conv_lstm layer in yolo_v3_tiny_lstm.cfg works for me:
[conv_lstm]
batch_normalize=1
size=3
pad=1
output=128
peephole=0
bottleneck=1
activation=leaky
@AlexeyAB I'm trying to train one of the YoloV3 LSTM models on #3114 But with all of them I got the same error:
My config is that one with the last repo ( but I also tried with August 15 repo and with cuda 10.0) I add that every model without LSTM works great (train and infer), yoloV3 or YoloV4, sam, pan... with the same dataset. yolo_v3_tiny_lstm.cfg.txt train.txt images samples : sample.zip obj.names.txt obj.data.txt
I am not the only one encountering issue at training lstm : #6531, #6708 What should we do to solve this?