search

bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码，可以用于训练自己的模型。
MIT License
305 stars 61 forks source link

训练日志 pr输出 指定训练gpu #11

Open dpk3 opened 3 years ago

dpk3 commented 3 years ago

因为我在服务器上运行这个程序，但是没有看到设定日志保存的地方 只看到了权重文件的保存 计算map可以指定阈值（如0.5）下的recall precision 最后 没有发现可以指定gpu的地方 负责一旦运行程序 所有gpu都得工作 影响别人的工作，请大神指点

dpk3 commented 3 years ago

关于指定GPU：若想使用gpu1和gpu2，程序开头多加一条语句 os.environ【"CUDA_VISIBLE_DEVICES"】 = "1,2"，作用是只允许gpu1和gpu2可用，然后使用Model = nn.DataParallel(Model, device_ids=【0,1】)，作用是从可用的两个gpu中搜索第0和第1个位置的gpu。
经测试 可行

bubbliiiing commented 3 years ago

好的，因为我没有多GPU所以我没有指定（全是单卡），日志保存是什么东西0 0