search

PaddlePaddle / PaddleClas

A treasure chest for visual classification and recognition powered by PaddlePaddle
Apache License 2.0
5.42k stars 1.16k forks source link

refine drop_path #3136

Closed wanghuancoder closed 4 months ago

wanghuancoder commented 4 months ago

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

paddle-bot[bot] commented 4 months ago

Thanks for your contribution!