Jun, 2024

快速优化器基准测试

TL;DR我们介绍了快速优化器基准(FOB),这是一种用于在其开发过程中评估深度学习优化器的工具。该基准支持来自多个领域(如计算机视觉、自然语言处理和图学习)的任务。其重点在于方便使用,具有人类可读的 YAML 配置、SLURM 集成和绘图实用程序。FOB 可与现有的超参数优化(HPO)工具一起使用,因为它处理训练和恢复运行。模块化设计使其能够集成到自定义流水线中,只需将其作为任务集合使用。我们通过一个优化器比较的示例展示了我们的工具的用法。FOB 可以在 GitHub 上找到:https://github.com/。