Блог

Я совершенно новичок в cntk. Недавно я установил cntk 2.7 (версия GPU) на свой ПК (Windows 10, процессор i5-7200U) с графическим процессором GeForce 940MX. Я пытаюсь настроить cntk и более быстрый пример обнаружения объекта rcnn, приведенный по ссылке нижеhttps://learn.microsoft.com/en-us/cognitive-toolkit/object-detection-using-faster-r-cnn

Я пытаюсь запустить игрушечный пример.

После запуска install_data_and_model.py в папке Examples/Image/Detection/FastRCNN

Я запускаю run_faster_rcnn.py Папка Examples/Image/Detection/FasterRCNN

Я получаю следующую ошибку:

 Selected GPU[0] GeForce 940MX as the process wide default device.

About to throw exception 'Failed to parse Dictionary from the input stream.'
Traceback (most recent call last):
File "run_faster_rcnn.py", line 34, in 
trained_model = train_faster_rcnn(cfg)

File "C:UsersHP-PCAnaconda3Libsite-packagescntkExamplesImageDetectionFasterRCNNFasterRCNN_train.py", line 291, in train_faster_rcnn
eval_model = train_faster_rcnn_e2e(cfg)

File "C:UsersHP-PCAnaconda3Libsite-packagescntkExamplesImageDetectionFasterRCNNFasterRCNN_train.py", line 314, in train_faster_rcnn_e2e
loss, pred_error = create_faster_rcnn_model(image_input, roi_input, dims_node, cfg)

File "C:UsersHP-PCAnaconda3Libsite-packagescntkExamplesImageDetectionFasterRCNNFasterRCNN_train.py", line 177, in create_faster_rcnn_model
base_model = load_model(cfg['BASE_MODEL_PATH'])

File "C:UsersHP-PCAnaconda3libsite-packagescntkinternalswig_helper.py", line 69, in wrapper
result = f(*args, **kwds)

File "C:UsersHP-PCAnaconda3libsite-packagescntkopsfunctions.py", line 1721, in load_model
return Function.load(model, device, format)

File "C:UsersHP-PCAnaconda3libsite-packagescntkinternalswig_helper.py", line 69, in wrapper
result = f(*args, **kwds)

File "C:UsersHP-PCAnaconda3libsite-packagescntkopsfunctions.py", line 1635, in load
return cntk_py.Function.load(str(model), device, format.value)

RuntimeError: Failed to parse Dictionary from the input stream.


[CALL STACK]
> CNTK::Internal:: UseSparseGradientAggregationInDataParallelSGD

- CNTK::operator>>

- CNTK::Function:: Load

- PyInit__cntk_py (x2)

- PyCFunction_Call

- PyEval_GetFuncDesc

- PyEval_EvalFrameEx (x2)

- PyFunction_SetAnnotations

- PyObject_Call

- PyEval_GetFuncDesc

- PyEval_EvalFrameEx (x2)

- PyEval_GetFuncDesc (x2)
  

Может кто-нибудь помочь мне с тем, в чем проблема?

Эта ошибка всегда возникает при завершении работы cntk при сохранении последней модели, поэтому разрыв файла модели