NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-23 16:54:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Update run.py

Browse source
This commit is contained in:
crStiv 2026-01-15 16:06:43 +02:00 • committed by GitHub
parent 20992ed5d5
commit 941fadd73c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/engine/run.py
View file

@ -220,7 +220,7 @@ class Run:
gradient_clip_algorithm=self.engine.gradient_clip_alg,
precision=precision_with_fallback(self.engine.precision), # type: ignore
accelerator=self.engine.accelerator,
log_every_n_steps=self.engine.logging_inteval,
log_every_n_steps=self.engine.logging_interval,
)
def get_tester(self) -> Trainer:
@ -333,7 +333,7 @@ class Run:
)
if self.engine.log_extra:
self._callbacks["extra"] = LogTrainingStats(
log_every_n_steps=self.engine.logging_inteval,
log_every_n_steps=self.engine.logging_interval,
**(
self.engine.log_extra
if isinstance(self.engine.log_extra, dict)