массив предсказаний пуст Маска RCNN

Question

Я работаю над маской RCNN, и мой массив прогнозов пуст. Я думаю, это связано с памятью. Это мой код, в котором я вызываю прогнозы:

model.eval()
with torch.no_grad():
prediction = model([img.to(device)], [target])

Когда я нахожусь в режиме отладки и оцениваю модель выражения ([img.to (device)], [target]), я получаю следующее уведомление:

{RuntimeError}[enforce fail at CPUAllocator.cpp:64] . DefaultCPUAllocator: can't allocate memory: you 
tried to allocate 172738321924 bytes. Error code 12 (Cannot allocate memory). But when I run the code 
as a whole I do not get any errors, except that the predictions are empty.