tensor2tensor icon indicating copy to clipboard operation
tensor2tensor copied to clipboard
verified publisher icon tensorflow

Fix resume checkpoint for tf2 codepath

Open almostimplemented opened this issue 3 years ago • 1 comments

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 avatar May 26 '22 10:05 almostimplemented

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

almostimplemented avatar May 26 '22 10:05 almostimplemented