My jetson nano keeps on hanging. After awhile it gives me the low memory warning and soon after my computer freezes. This only happens when running the code and model below