Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

training_utils: export neptune config and mixin

Browse source 
Like W&B. With this, #371 would have broken `test_import_training_utils`.

Follow up of #371 and #372.
This commit is contained in:
Cédric Deltheil 2024-05-29 07:40:18 +00:00 committed by Cédric Deltheil
parent 7dba1c8034
commit ddcd84c740
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/refiners/training_utils/__init__.py
View file

@ -41,6 +41,7 @@ from refiners.training_utils.config import (
Optimizers,
TrainingConfig,
)
from refiners.training_utils.neptune import NeptuneConfig, NeptuneMixin
from refiners.training_utils.trainer import Trainer, register_callback, register_model
from refiners.training_utils.wandb import WandbConfig, WandbMixin
@ -54,6 +55,8 @@ __all__ = [
"CallbackConfig",
"WandbMixin",
"WandbConfig",
"NeptuneMixin",
"NeptuneConfig",
"LRSchedulerConfig",
"OptimizerConfig",
"TrainingConfig",