search

Sierkinhane / CRNN_Chinese_Characters_Rec

(CRNN) Chinese Characters Recognition.
1.81k stars 537 forks source link

怎么使用多卡训练 #311

Open TinkingLoeng opened 2 years ago

sunjunlishi commented 7 months ago

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

sunjunlishi commented 6 months ago

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

    ps = preds.size()

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