Checkpoints are currently pickled specific to the python version, so running on NCC (python 3.8), cannot be run on local version (python 3.10)