在终端运行:
python trainval_net.py --cuda
在pycharm中
parser.add_argument('--cuda', dest='cuda', help='whether use CUDA', action='store_true')
修改成:
parser.add_argument('--cuda', dest='cuda', default=True,
help='whether use CUDA')