- 博客(4)
- 收藏
- 关注
原创 yolov4-tiny网络结构
yolov4-tiny中,有个tebie之处就是,route层有groups和group_id,我认为就是,将输入数据按照通道个数平均分成groups分,实际数据的就是第group_id份数据。
2020-07-31 14:32:15 7667 3
原创 多GPU实现同时训练模型
1.1单GPU:import osos.environ[“CUDA_VISIBLE_DEVICES”] = “0”说明:设置单块GPU编号1.2多GPU:device_ids = [0,2,4,6]#说明:GPU编号,一块显卡集成两块GPU,则同一块卡的编号不能同时出现。model = nn.DataParallel(model, device_ids=device_ids).cud...
2019-03-20 17:22:38 2156
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人