• Derek Chow's avatar
    Rename Saver used in init_fn · ec91f6b8
    Derek Chow authored
    Bug:
    We use two different Saver objects during training, one to initialize from a
    checkpoint and another to save and restore from checkpoints during training.
    Both these Savers were bound to same variable name causing init_fn function
    to use the latter instead of the former when called.
    ec91f6b8
trainer.py 12.3 KB