Open TinkingLoeng opened 2 years ago
hi,这个耗时呀。使用双卡可以提升一倍的速度呀。怎么用呀,loss的时候 那个问题没法解决。怎么解决呀
简单的尝试解决,更改维度: preds = model(inp).cpu()
ps = preds.size() new_shape = ((int)(ps[0]/2), ps[1]*2, ps[2]) preds = preds.view(*new_shape)
hi,这个耗时呀。使用双卡可以提升一倍的速度呀。怎么用呀,loss的时候 那个问题没法解决。怎么解决呀