sgflower66

Multi-gpu support---only can run on gpu 0

2

when I run the code, although I use multi-gpu, only gpu 0 is used. I use config below: python ./train.py \ --gpu_ids 0,1,2,3 --batchsize 16 ............

 I trained the model by your newest code (no change), and used my own dataset which 'Dataset A' are clothing edges( extracted by hed), 'Dataset B' are original clthothing...

RuntimeError: merge_sort: failed to synchronize: an illegal memory access was encountered.When train by myself with "./experiments/scripts/train_faster_rcnn.sh 0 pascal_voc vgg16"

3

After Loading initial model weights from data/imagenet_weights/vgg16.pth Loaded. It will appear this problem:(,(I didn't change anything,just try to train it) File "/home/hdd2/pytorch-faster-rcnn/tools/../lib/nets/network.py", line 87, in _proposal_layer self._feat_stride, self._anchors, self._num_anchors) File...