Very high GPU memory usage

[vaishnavm217]

Hi,
I have been running the model using test.py and modified it run multiple files. The GPU memory keeps on increasing,from 3gigs to 9 gigs.
Is this due to poor garbage collection?

[DCNSW]

@clcarwin

[ethanyhzhang]

@vaishnavm217 in the detect() function, the img of numpy is transformed to torch Variable, as the code shows : img = Variable(torch.from_numpy(img).float(), valotile=True).cuda()
you might notice that valotile is deprecated in the latest version, so just change the code as below to see if it helps:
with torch.no_grad():
img = Variable(torch.from_numpy(img).float()).cuda()
olist = net(img)