Ah, OK, I think the issue was I was only using norb_small.FoveatedNORB. I had forgotten that was a different class.