怎么使用多卡训练

[TinkingLoeng]

[sunjunlishi]

hi，这个耗时呀。使用双卡可以提升一倍的速度呀。怎么用呀，loss的时候 那个问题没法解决。怎么解决呀

[sunjunlishi]

简单的尝试解决，更改维度： preds = model(inp).cpu()

    ps = preds.size()
    
    new_shape = ((int)(ps[0]/2), ps[1]*2, ps[2])
    preds = preds.view(*new_shape)