Fix resume checkpoint for tf2 codepath

tensorflow / tensor2tensor

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

Apache License 2.0

15.58k stars 3.51k forks source link

Open almostimplemented opened 2 years ago

almostimplemented commented 2 years ago

Cheap, localized solution to the bug described in #1911

For the TF2 codepath, provide additional two methods to the DummyModule: load_checkpoint and get_trainable_variables.

almostimplemented commented 2 years ago

Hm, maybe these can be added directly to the DummyModule construction, down at def framework (line 83 in original).