The dumped model can be obtained by running xornet.py
ModuleNotFoundError: No module named ‘megbrain’
The dumped model can be obtained by running xornet.py
ModuleNotFoundError: No module named ‘megbrain’
您好,麻烦提供一下是从哪里看到的这个内容呢?请提供一下链接,方便我们排查一下。
感谢您关注 MegEngine 项目:)
1.你好,我在测试MegEngine/sdk/xor-deploy/xor_deploy.cpp这个文件,LD_LIBRARY_PATH=$MGE_INSTALL_PATH:$LD_LIBRARY_PATH ./xor_deploy xornet_deploy.mge 0.6 0.9
发现需要xornet_deploy.mge模型文件,在MegEngine/sdk/load-and-run/README.md
Obtain the model file by running xornet.py,没有找到xornet.py所在目录和脚本
2.MegEngine/sdk/load-and-run/dump_with_testcase_mge.py 脚本报错,import megbrain as mgb,没有这个包
文章上半部分提到,训练好的异或网络模型在项目里写到了。
里面具体提到了 xor_net 的定义
import megengine.module as M
import megengine.functional as F
class XORNet(M.Module):
def __init__(self):
self.mid_dim = 14
self.num_class = 2
super().__init__()
self.fc0 = M.Linear(self.num_class, self.mid_dim, bias=True)
self.fc1 = M.Linear(self.mid_dim, self.mid_dim, bias=True)
self.fc2 = M.Linear(self.mid_dim, self.num_class, bias=True)
def forward(self, x):
x = self.fc0(x)
x = F.tanh(x)
x = self.fc1(x)
x = F.tanh(x)
x = self.fc2(x)
return x
xor_net = XORNet()
好的,谢谢回答!