def _calculate_max_updates(self):
config_max_updates = self.training_config.max_updates
config_max_epochs = self.training_config.max_epochs
max_updates, _ = get_max_updates(
config_max_updates,
config_max_epochs,
self.train_loader,
self.training_config.update_frequency,
)
return max_updates
def _calculate_max_updates(self) -> None:
self._max_updates = self.trainer_config.max_steps
self._max_epochs = self.trainer_config.max_epochs
if self._max_updates is None and self._max_epochs is None:
raise ValueError(
"Neither max_updates nor max_epochs is specified.")
self._max_updates, max_epochs = get_max_updates(
self._max_updates,
self._max_epochs,
self.train_loader,
self.trainer_config.accumulate_grad_batches,
)
self._max_epochs = math.ceil(max_epochs)
return self._max_updates