Need to make sure that if an agent with a wrapper is saved on a GPU, it can be gracefully loaded on a CPU, and the other way around.
Need to make sure that if an agent with a wrapper is saved on a GPU, it can be gracefully loaded on a CPU, and the other way around.