匯出視覺化圖

import sys
sys.path.append('/home/')
from models.SelfNestingNet import *
from my_utils import *

net = mx.sym.Variable('data')
net = SelfNestingNet(nClasses=10,basicBlockType ='BasicUnit' )
mx.viz.plot_network(net(mx.sym.var("img")))
digraph = mx.viz.plot_network(net, shape={"data":(1, 1, 32, 32
 
)},node_attrs={"fixedsize":"false"})
digraph.view()

錯誤：

Symbol only support integer index to fetch i-th output

ndarray, x = nd.random.uniform(shape=(1,3,224,224),ctx=mx.gpu())

這個是MXNet根據動態圖的一次forward計算, 生成靜態圖的流程問題. 可以理解成:

1.hybridize(), 表示程式希望生成高效的靜態圖模型; 2.net(mxnet.ndarray.xxx()), 在這一次forward中程式能計算其所需的資源, 並寫入到靜態圖的資訊中; 3.再export()