如何想象Caffe网中的不同阶段？

Question

我正在与Caffe合作开发一体化网络。我已经弄清楚如何使用阶段来分离火车+测试和部署。但是，我想要想象阶段。

目前，我在Jupyter笔记本中获得了以下代码（旧版本，在此处复制了一些代码，考虑了列车和部署的不同prototxt文件）

from google.protobuf import text_format
from IPython import display
import caffe.draw

trained_net = caffe.proto.caffe_pb2.NetParameter()
text_format.Merge(open('train_val.prototxt').read(), trained_net)

display.Image(caffe.draw.draw_net(trained_net, 'BT'))

我的新train_val_deploy.prototxt file中的某些图层包含

include {
     stage: "create"
}

和

include {
     stage: "deploy"
}

我如何绘制这些阶段？