Currently Engine implicitly assumes :
|
_state_dict_all_req_keys = ("epoch_length", "max_epochs") |
but we recently introduced possibility to trigger a run with max_iters. If epoch length is unknown, saving and reloading engine's state probably wont work. We have to ensure that everything still works as expected.
@thescripted would you be interested in looking into this ?