feat: add max_steps to training

Commit

1 year ago

feat: add max_steps to training alternative to max_epochs which is helpful when running g experiments with varying amounts of data