mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-21 13:48:46 +00:00
Fix clock step inconsistencies on batch end
This commit is contained in:
parent
09af570b23
commit
bbb46e3fc7
|
@ -179,10 +179,8 @@ class TrainingClock(Callback["Trainer[BaseConfig, Any]"]):
|
||||||
self.log(f"Iteration {trainer.clock.iteration} started.")
|
self.log(f"Iteration {trainer.clock.iteration} started.")
|
||||||
self.log(f"Step {trainer.clock.step} started.")
|
self.log(f"Step {trainer.clock.step} started.")
|
||||||
|
|
||||||
def on_batch_end(self, trainer: "Trainer[BaseConfig, Any]") -> None:
|
|
||||||
self.log(f"Step {trainer.clock.step} ended.")
|
|
||||||
|
|
||||||
def on_backward_end(self, trainer: "Trainer[BaseConfig, Any]") -> None:
|
def on_backward_end(self, trainer: "Trainer[BaseConfig, Any]") -> None:
|
||||||
|
self.log(f"Step {trainer.clock.step} ended.")
|
||||||
trainer.clock.step += 1
|
trainer.clock.step += 1
|
||||||
trainer.clock.num_batches_processed += 1
|
trainer.clock.num_batches_processed += 1
|
||||||
trainer.clock.num_minibatches_processed += 1
|
trainer.clock.num_minibatches_processed += 1
|
||||||
|
|
|
@ -66,7 +66,7 @@ def scoped_seed(seed: int | Callable[..., int] | None = None) -> Callable[..., C
|
||||||
actual_seed = seed(*args) if callable(seed) else seed
|
actual_seed = seed(*args) if callable(seed) else seed
|
||||||
seed_everything(seed=actual_seed)
|
seed_everything(seed=actual_seed)
|
||||||
result = func(*args, **kwargs)
|
result = func(*args, **kwargs)
|
||||||
logger.debug(f"Restoring previous seed state")
|
logger.trace(f"Restoring previous seed state")
|
||||||
random.setstate(random_state)
|
random.setstate(random_state)
|
||||||
np.random.set_state(numpy_state)
|
np.random.set_state(numpy_state)
|
||||||
torch.set_rng_state(torch_state)
|
torch.set_rng_state(torch_state)
|
||||||
|
|
Loading…
Reference in a new issue