CUDA_VISIBLE_DEVICES=$GPU_ID
:指明所使用的GPU ID,$GPU_ID
需修改为指定ID--dataset pascal_voc --net res101
:在pascal_voc上使用resnet101进行训练--bs $BATCH_SIZE --nw $WORKER_NUMBER
:可以根据GPU内存大小来自适应地设置BATCH_SIZE(批处理尺寸)和WORKER_NUMBER。在具有12G内存的Titan Xp上,它最多可以为4--lr $LEARNING_RATE --lr_decay_step $DECAY_STEP
:**学习率(Learning rate)**作为监督学习以及深度学习中重要的超参,其决定着目标函数能否收敛到局部最小值以及何时收敛到最小值。合适的学习率能够使目标函数在合适的时间内收敛到局部最小值。RuntimeError: The expanded size of the tensor (2) must match the existing size (0) at non-singleton dimension 1
AttributeError: 'int' object has no attribute 'astype'
/lib/roi_data_layer/roibatchLoader.py
中第52行的target_ratio = 1
改为target_ratio = np.array(1)
AttributeError: 'Tensor' object has no attribute 'index'