search

MegEngine / MegFlow

Efficient ML solution for long-tailed demands.
Apache License 2.0
402 stars 42 forks source link

pytorch 模型怎么dump成mge模型 #49

Open zhangshitoday opened 2 years ago

zhangshitoday commented 2 years ago

请问用pytorch训的resnet18 二分类模型怎么得到mge模型呀

cqchu commented 2 years ago

能了解一下这件事的最终目的或者更多的细节信息吗？ 如果只是单纯转换可以使用torch2mge的转换器，https://github.com/MegEngine/torch2mge

cqchu commented 2 years ago

需要在转换器的第46行处增加两行代码，如下： if tk == "compute_mode" or tk == "param_dim": continue

修改后效果如 https://github.com/cqchu/torch2mge/blob/master/convert.py#L46 所示