Recently, the lack of pytorch cuda video memory occurred in the test stage
created at 11-22-2021
views: 2
It shouldn't happen in the test phase, because at this time, the derivation is no longer needed. The inspection found that there is less time in the model calculation.
with torch.no_grad:
That should:
with torch.no_grad():
pred = model(input)
