search

zhy520xp / mace-makefile-project

makefile 交叉编译 libmace.a，并能在嵌入式端调用GPU来跑深度学习模型
96 stars 18 forks source link

如何将mace更新到最新的版本。 #12

Open leeburt opened 6 years ago

leeburt commented 6 years ago

首先感谢你的分享。我在测试的过程中发现0.9版本转换出来的gpu版模型在运行的时候报错，cpu版本运行没有问题，怀疑是版本不兼容问题。目前版本是0.8如何更新到最新的版本0.9 或者以后推出来的版本。

zhy520xp commented 6 years ago

应该是0.9版本相对于0.8版本，GPU实现代码有改动导致的。因为Mace是使用codegen自动生成GPU部分的代码，而我这个工程是把这部分自动生成的代码包含进去了得。所以0.9版本自动生成的代码跟我现在那部分代码不兼容。。。要解决这个问题，你需要把我源码里的自动生成那部分代码替换掉。。。这个确实比较麻烦。。。

leeburt commented 6 years ago

能具体给提示一些信息吗，或者有什么资料可以学习参考。

zhy520xp commented 6 years ago

为什么非要用0.9版本呢？0.8版本不是也能用吗。如果你真想要去改的话，你把这份代码和mace最新的代码对比一下，把不相同的部分改掉就行。但是这样做对能力要求比较高，你要比较熟悉mace源码框架才行

leeburt commented 6 years ago

就是想学习一下。我去试一试。

leeburt commented 6 years ago

请问一下，codegen/engine/mace_engine_factory.h这个文件是怎样生成的。def gen_mace_engine_factory(model_tags, template_dir, embed_model_data, output_dir): 这个函数的几个参数分别代表什么意思？