增强您的 JAX/Optax 训练：轻松调整学习率！

本文揭示了一种巧妙的技术，使用 `optax.inject_hyperparams` 在 JAX/Optax 模型训练期间动态修改学习率。 这种创新方法允许实时调整并简化调试，相对于传统方法具有显著优势。 它赋予研究人员和开发人员更大的训练过程控制权。

• •`optax.inject_hyperparams` 允许您将学习率等超参数视为优化器状态的一部分。
• •这使得在训练期间直接修改学习率成为可能。
• •非常适合调试和尝试不同的学习策略。