allow to save in cpu when run in gpu
pickle in cpu