Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 15:02:01 +00:00
Code Issues Actions 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 enum import Enum
from logging import warn from logging import warning
from pathlib import Path from pathlib import Path
from typing import Annotated, Any, Callable, Iterable, Literal, Type, TypeVar from typing import Annotated, Any, Callable, Iterable, Literal, Type, TypeVar
@ -132,7 +132,7 @@ class OptimizerConfig(BaseModel):
) )
case Optimizers.Prodigy: case Optimizers.Prodigy:
if self.learning_rate != 1.0: 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( return Prodigy(
lr=self.learning_rate, lr=self.learning_rate,
params=params, params=params,