在英特尔神经计算stick2上训练detectron2模型

问题描述

有谁知道如何在英特尔神经计算棒2设备上训练基于pytorch的模型detectron2模型动物园的faster_rcnn_R_50_FPN_3x模型?
我已经安装了 openvino-toolkit 并在我的设备上正确运行了他们的一个演示。
我已经在 faster_rcnn_R_50_FPN_3x 上使用detectron2 在自定义数据集上训练了 google-colab-gpu,现在我必须在英特尔神经计算 stick2 vpu 上训练它以比较 AP 结果

解决方法

您无法在英特尔神经计算棒 2 (NCS2) 上训练模型。 NCS2 将与 OpenVINO™ 工具包一起使用。 OpenVINO™ 工具包由两个主要组件组成,即模型优化器和推理引擎。推理引擎 MYRIAD 插件已开发用于英特尔® 神经计算棒 2 上的神经网络推理。请查看此 OpenVINO™ Toolkit Workflow