Allow to configure checkpoint_low_metric and checkpoint_high_metric for training.Loop

google / trax

Trax — Deep Learning with Clear Code and Speed

Apache License 2.0

8.07k stars 813 forks source link

Allow to configure checkpoint_low_metric and checkpoint_high_metric for training.Loop #1659

Closed manifest closed 3 years ago

manifest commented 3 years ago

Passing checkpoint_low_metric and checkpoint_high_metric to training.Loop from train_lib.train allows using gin-config to control these arguments.