@Dejan_Grubisic I am seeing the same error when restoring a checkpoint for a 3-layer network. However, if my network has only 2 layers, the restore works fine. See my post at ValueError when restoring checkpoint with PPO. I suspect you are seeing the same issue I was. I’m a little sleepy, but is it possible that your restore using checkpoint_path is loading a network with different structure than what is defined in best_config?