We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9699a88 commit 0479b7eCopy full SHA for 0479b7e
src/fmcore/framework/_trainer/LocalTrainer.py
@@ -10,6 +10,7 @@
10
11
from bears import FileMetadata
12
from bears.util import String, Timer, safe_validate_arguments
13
+from pydantic import model_validator
14
15
from fmcore.framework._algorithm import Algorithm
16
from fmcore.framework._dataset import Dataset, Datasets, DataSplit
@@ -24,6 +25,12 @@ class LocalTrainer(Trainer):
24
25
def initialize(self, **kwargs):
26
pass
27
28
+ @model_validator(mode="before")
29
+ @classmethod
30
+ def local_trainer_params(cls, params: Dict) -> Dict:
31
+ params: Dict = cls._set_common_trainer_params(params)
32
+ return params
33
+
34
@staticmethod
35
def local_logger(text: str, verbosity: int, tracker: Tracker):
36
pid: int = os.getpid()
0 commit comments