xornet.py为找到,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,没有这个包

这边的 README 跟我们的内部版本有些搞混了,请参考这里:https://megengine.org.cn/doc/latest/advanced/deployment.html

,也没看到在哪啊

文章上半部分提到,训练好的异或网络模型在项目里写到了。

里面具体提到了 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()

好的,谢谢回答!