pytorch
pytorch训练模型的内存一直增加,在我的问题是:
创建了多个dataloader,但是训练的时候只用了一个dataloader的数据,但是另外一个dataloader也参与了循环,具体看下面的例子:
dataloaderA =
dataloaderB =
for i, data in enumerate(zip(dataloaderA, dataloaderB)):
data = data[0]
input = data[0]
label = data[1]
这样就只用了dataloaderA的数据,没有用dataloaderB的数据