Cannot load hyper parameters properly from a checkpoint

[alexcoca]

Guess using the colab to reproduce it will help the developers iron it out quickly!