zhenghao977/FCOS-PyTorch

不能使用多卡训练

Opened this issue · 1 comments

修改这行代码后仍然只使用编号为“1”的gpu进行训练parser.add_argument("--n_gpu", type=str, default='0,1', help="number of pu threads to use during batch generation")