如何解决HybridSequential.hybridize返回无
我这样建立了一个网络
sns.kdeplot(data=data,x='x',y='y')
然后我尝试使用张量板将其可视化
net = nn.HybridSequential()
# Add a sequence of layers.
net.add(
nn.Conv2D(channels=64,kernel_size=2,strides=2,groups=1,activation='relu'),#1
nn.BatchNorm(),nn.Conv2D(channels=256,kernel_size=4,groups=2,#2
nn.BatchNorm(),kernel_size=3,padding=1,strides=1,#3
nn.BatchNorm(),#4
nn.BatchNorm(),nn.Dense(10)
)
但是net.hybridize()返回None。
这就是add_graph返回NoneType参数错误的原因。
解决方法
net.hybridize()不返回任何内容,因此返回错误。
您需要做的是在sw.add_graph(net)之前杂交并通过网络传递样本数据
# initialize the weights unless you did it already
net.initialize()
# hybridize
net.hybridize()
# pass some data (adjust shapes accordingly)
y = net(mx.nd.ones([1,16,100,100])) # batch x channels x H x W
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。