Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 10:55:27 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix deprecation warning
Some checks failed
CI / lint_and_typecheck (push) Has been cancelled
Details
Deploy docs to GitHub Pages / Deploy docs (push) Has been cancelled
Details
Spell checker / Spell check (push) Has been cancelled
Details

Browse source
This commit is contained in:
Pierre Chapuis 2024-08-28 10:39:41 +02:00
parent e643006887
commit c8ff6d95f0
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/training_utils/config.py
View file

@ -1,5 +1,5 @@
from enum import Enum
from logging import warn
from logging import warning
from pathlib import Path
from typing import Annotated, Any, Callable, Iterable, Literal, Type, TypeVar
@ -132,7 +132,7 @@ class OptimizerConfig(BaseModel):
)
case Optimizers.Prodigy:
if self.learning_rate != 1.0:
warn("Prodigy learning rate is not 1.0, this might cause instability.")
warning("Prodigy learning rate is not 1.0, this might cause instability.")
return Prodigy(
lr=self.learning_rate,
params=params,