witnesslyq的博客

原创 训练网络时，GPU占用率一直很低

关于GPU占用率的问题,两部分代码都在下面了，不知道哪里有问题，大家帮我看看，什么numworks,pin_memory都试过没用。