Как я могу проверить быстрые веса RCNN во время тренировок на глюоне? - PullRequest
0 голосов
/ 19 апреля 2020

Я пытаюсь обучить более быстрой области сверточного NN, следуя этому примеру. https://gluon-cv.mxnet.io/build/examples_detection/train_faster_rcnn_voc.html

Это то, что я сделал до сих пор, но net не может обнаружить объекты. https://drive.google.com/file/d/1WevimDyTP1lvq_A0OBRMgC-PH8pK4iBv/view?usp=sharing

Я хочу проверить, обновляются ли веса net, моя идея - печатать их в каждую эпоху, но я не смог получить доступ к их.

Я пытался:

print(net.weights)
print(net.collect_params().data)
print(rpn_loss1)
print(rpn_loss2)
print(rcnn_loss1)
print(rcnn_loss2)

и еще немного, я не могу вспомнить

...