RuntimeError: tensors are on different GPUs, but I just run “CUDA_VISIBLE_DEVICES=1”. Why? Even I just run
python train.py
this error still exists.
I cut the conv1_1 to conv3_3 layers of vgg16 to build my model.