Loading weights for CPU model while trained on GPU

[fmassa (Francisco Massa)]

This is a different issue, and is related to [solved] KeyError: 'unexpected key "module.encoder.embedding.weight" in state_dict'