Enhancement

  • 使用 MegEngine 的随机分布生成函数替换 numpy,加速参数初始化

Bug Fix

  • element-wise 函数增加数值输入参数的支持
  • collator挪到data_collection_process里,在divide模式下不用额外写collate的逻辑
  • 移除_ParallelDataLoaderIter内部的strict属性,在任意worker发生异常时可直接抛出异常
  • 内部的随机数由np.random.RandomState替换成megengine.randon.rng模块生成,保证数据供给的可复现性
  • 修改 load_and run README.md 的内部代码,提供 xornet.py 示例
  • 修复 ImageNet 解压错误
  • 修复 python3.7下多进程报错的问题[TypeError: can’t pickle weakref objects]
  • 修复 InternalGraphGenerator 拼写错误
  • 修复 batchnorm、loss、tranform、tensor API文档问题

Enhancement

  • Speed up parameter initialization by using MegEngine's random distribution generator function instead of NumPy.

BugFix

  • support scalar inputs in element-wise functions
  • Remove internal codes in README.md and add missed example file xornet.py
  • Move collator to data_collection_process, no need to write additional collate in divide mode
  • Remove the strict attribute inside _ParallelDataLoaderIter, and throw an exception directly when any worker exception occurs
  • The random number generation module is replaced by np.random.RandomState to megengine.randon.rng to ensure the reproducibility of data supply
  • Fix imagenet extraction error
  • Fix DataLoader error for Multi-process under python3.7
  • Fix typo InternalGraphGenerator
  • Fix docstring for batchnorm、loss、tranform、tensor

项目简介

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

🚀 Github 镜像仓库 🚀

源项目地址

https://github.com/MegEngine/MegEngine

发行版本 37

MegEngine v1.13.1

全部发行版

贡献者 39

全部贡献者

开发语言

  • C++ 79.8 %
  • Cuda 13.8 %
  • Python 4.9 %
  • C 0.9 %
  • CMake 0.5 %