Matteo Dunnhofer
Matteo Dunnhofer
Hi @chutongz have you already tried to lower the learning rate? Like to 1e-4?
Hi @iodncookie, sorry for the late reply! Are you sure to have loaded the right checkpoint of the model?
Hi @iodncookie Are you training and testing a different model than the AlexNet present in the repo? I noticed this line in the code you pasted `_, pred = alexnet.darknet53_classifier(x,False)`...
Have you tried to train AlexNet? Does it give you the same problem?