fix gpu pickled files to run on cpu