Created by: TomorrowIsAnOtherDay
Users are not expected to specify the place where params are placed(CPU or GPU).
In our framework , we encourage users to specify the GPU to use by setting the environment variable like export CUDA_VISIBLE_DEVICES="0"
, or to use the CPU to store parameters by export CUDA_VISIBLE_DEVICES=""