本文章分享一下自身遇到的一个问题,参考很多其他文章对GPU利用率低的解答发现自己都解决了,可是利用率还是一会儿10%一会儿90%。我的原因是使用了VGG模块用于计算感知损失。
for epochinrange(start_epoch, opt.nEpochs):for iteration, batchinenumerate(training_data_loader,1):...
vgg= Vgg19(requires_grad=False).to('cuda')...
正确的做法是把感知损失的加载放在epoch外面,加载一次即可。