Generator checkpoint not found

[ahmed-kar]

When trying to train it, despite the latest.pt file being in the checkpoints folder it always finishes with Generator Checkpoint not found, It seems as the generator checkpoint is giving back None in model_step. Also checkpoints are only saved on debug CPU mode, not the normal GPU mode.
Also why does the model not learn despite making the prefill steps really large, and the issue after the prefill steps are completed the program just exits, again because it seems like checkpoint is not found. These issues are not dependency issues so it seems.

Thank you