justchenhao / STANet

official implementation of the spatial-temporal attention neural network (STANet) for remote sensing image change detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问要怎么多显卡进行训练

Qingwang-7 opened this issue · comments

我调了好久,都没成功

看options中的base_options.py 中的参数 --gpu_ids

看options中的base_options.py中的参数--gpu_ids

我设置为--gpu_ids 0,3 只能使用第一张显卡

看options中的base_options.py中的参数--gpu_ids

我设置为--gpu_ids 0,3 只能使用第一张显卡

我也是,设置了多张,却还是只能用第一张显卡

commented

ds设置成1就不能跑,爆显存,最多只能设成ds=2,是不是得设成多卡模型并行来训练

看options中的base_options.py中的参数--gpu_ids

我设置为--gpu_ids 0,3 只能使用第一张显卡

我也是,设置了多张,却还是只能用第一张显卡

请问你解决了吗?我也是这个问题