Open
Description
For learning rate schedulers, there is a code that requires steps_per_epoch entry. The current code looks like this:
extra_logs[hyperparam] = float(value((epoch+1) * self.config_info["steps_per_epoch"])) ## here replace with self.params.get('steps')
Replace this part with a call that enables to get the steps_per_epoch term. See if self.params.get("steps") works.
Metadata
Metadata
Assignees
Labels
No labels