hello，i run the code and this error i can't resolve,and the i cant visualize with the port in your readme files.

[crazygirl1992]

RuntimeError: Assertion `THCTensor_(checkGPU)(state, 3, input, output, weight)' failed. Some of weight/gradient/input tensors are located on different GPUs. Please move them to a single one. at /opt/conda/conda-bld/pytorch_1535493744281/work/aten/src/THCUNN/generic/SpatialDepthwiseConvolution.cu:16