`pw_checkpoints` undefined in `train.py`

google-research / t5x

Apache License 2.0

2.65k stars 301 forks source link

`pw_checkpoints` undefined in `train.py` #1520

Open yhtang opened 7 months ago

yhtang commented 7 months ago

When do training, we hit the error:

  File "/opt/t5x/t5x/train.py", line 213, in train
    checkpoint_cfg.restore.checkpointer_cls, pw_checkpoints.Checkpointer
NameError: name 'pw_checkpoints' is not defined

The code path appears to exist for a long time but has not been run by default until recently.