openvino_notebooks: 226-yolov7-optimization on Ubuntu

When I run this notebook on Ubuntu with a successful setup of the virtual env and requirements.txt install…the kernel dies on my machine half way through every time…would you have tips to try?

Its this block of code towards the end…where it does run I can see the process go from 0 to 100% but after a 100% is met the Kernel dies and I cant make it any further.

mp, mr, map50, map, maps, num_images, labels = test(data=data, model=compiled_model, dataloader=dataloader, names=NAMES)
# Print results
s = ('%20s' + '%12s' * 6) % ('Class', 'Images', 'Labels', 'Precision', 'Recall', 'mAP@.5', 'mAP@.5:.95')
print(s)
pf = '%20s' + '%12i' * 2 + '%12.3g' * 4  # print format
print(pf % ('all', num_images, labels, mp, mr, map50, map))

Any options to try greatly appreciated.