Open dpk3 opened 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。
经测试 可行
好的,因为我没有多GPU所以我没有指定(全是单卡),日志保存是什么东西0 0
因为我在服务器上运行这个程序,但是没有看到设定日志保存的地方 只看到了权重文件的保存 计算map可以指定阈值(如0.5)下的recall precision 最后 没有发现可以指定gpu的地方 负责一旦运行程序 所有gpu都得工作 影响别人的工作,请大神指点