c907ac173e
* adding pre-commit auto-update GH action and bumping plugin versions * running updated pre-commit plugins * sorry to revert, but pylint complained * Update .pre-commit-config.yaml Co-authored-by: Wing Lian <wing.lian@gmail.com> --------- Co-authored-by: Dan Saunders <dan@axolotl.ai> Co-authored-by: Wing Lian <wing.lian@gmail.com>
17 lines
274 B
Python
17 lines
274 B
Python
"""
|
|
config args for grokfast plugin
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class GrokfastArgs(BaseModel):
|
|
"""
|
|
Input args for Grokfast optimizer.
|
|
"""
|
|
|
|
grokfast_alpha: Optional[float] = 0.98
|
|
grokfast_lamb: Optional[float] = 2.0
|