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

[dpk3]

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

[dpk3]

关于指定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]

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