【文件属性】:
文件名称:flaxOptimizersBenchmark:对标代码以评估亚麻优化器
文件大小:7KB
文件格式:ZIP
更新时间:2021-04-19 00:20:10
benchmarking deep-learning optimizer flax
亚麻优化基准
该存储库的目标是为用户提供一组评估Flax优化器的基准。
我们的目标是:
给出一些快速基准以快速评估新的优化器
提供经典的基准测试和绘图功能,以帮助希望发布新的优化程序的深度学习论文的作者
安装
您可以使用以下命令安装此librarie:
pip install git+https://github.com/nestordemeure/flaxOptimizersBenchmark.git
用法
快速基准
去做
基准测试慢
去做
亚麻优化器
包含Flax中大量优化器的实现。
包含我在亚麻中对Adahessian二阶优化器的实现。
包含官方的Flax优化器。
【文件预览】:
flaxOptimizersBenchmark-main
----.gitignore(2KB)
----setup.cfg(40B)
----README.md(957B)
----LICENSE(11KB)
----setup.py(962B)
----flaxOptimizersBenchmark()
--------__init__.py(52B)