DriessenA

Hi, I run into a similar issue. I save my checkpoints with metrics, train it only on CPU and then on the same machine I want to load a checkpoint....