Models icon indicating copy to clipboard operation
Models copied to clipboard

复现 Googlenet

Open xxr3376 opened this issue 5 years ago • 4 comments

任务描述

  • 复现 Googlenet ,训练正常收敛,验收指标符合预期,并将代码提交到 offical/vision/classification/models 下

目标

  • 数据集ImageNet
  • 准确率和论文一致或更高
  • 脚本可以完整完成训练步骤
  • 提供训练后的权重文件
  • 提交至 https://github.com/MegEngine/Hub

xxr3376 avatar Jul 03 '20 11:07 xxr3376

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

lijiansong avatar Apr 24 '21 12:04 lijiansong

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

xxr3376 avatar Apr 25 '21 07:04 xxr3376

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

你指的是 megengine.functional.topk 吗?

ArtShiba avatar May 07 '21 07:05 ArtShiba

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

lijiansong avatar May 07 '21 08:05 lijiansong