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