PaddleClas icon indicating copy to clipboard operation
PaddleClas copied to clipboard

refine drop_path

Open wanghuancoder opened this issue 9 months ago • 2 comments

drop_path将1个float转换为Tensor,使用了to_tensor。to_tensor会使用阻塞式的H2D拷贝,导致Python端阻塞,进而导致无法预加载kernel,导致GPU利用率低。 改为full可以解决这个问题。

wanghuancoder avatar Apr 28 '24 06:04 wanghuancoder

Thanks for your contribution!

paddle-bot[bot] avatar Apr 28 '24 06:04 paddle-bot[bot]

我看了下文档,测试了下。paddle.to_tensor(value)得到的是一个0-D Tensor,所以是否应该使用paddle.full(shape=[], value)这样?

TingquanGao avatar Apr 28 '24 06:04 TingquanGao