search

MegEngine / Models

采用MegEngine实现的各种主流深度学习模型
Other
303 stars 99 forks source link

复现 Googlenet #40

Open xxr3376 opened 4 years ago

xxr3376 commented 4 years ago

任务描述

目标

lijiansong commented 3 years ago

@xxr3376最近在试用meg-engine推出的大模型训练的dtr功能，发现model zoo里都没有vgg这种经典的模型:-) 感觉类似这种https://www.tensorflow.org/api_docs/python/tf/keras/applications模型官方repo都应该有啊

xxr3376 commented 3 years ago

@lijiansong VGG 这些模型实在有些古老，所以在复现的时候就没有首先选这些。欢迎大家给我们的 ModelHub 提交模型呀~

cheekyshibe commented 3 years ago

@xxr3376 还有个问题，目前的算子列表里没有topk这种算子，所以如果想实现基本的机器学习算法，如kNN、决策树、kmeans等这些算法，暂时做不了？

你指的是 megengine.functional.topk 吗？

lijiansong commented 3 years ago

@MegChai 是的，看了一下列表，还是挺完备的