I am using a NANO, which whilst i know is not the most powerful machine in the world but i keep getting out of memory errors whilst going through tutorials.
following this one: Transfer learning with TensorFlow Hub | TensorFlow Core
and I get OOM when allocating tensor with shape[32,3,224,224] at the prediction stage.
I have had loads of errors like that when trying to make my own models, which I kinda understood as I was making layers but I have a preloaded one here?
Jetpack 4.4.1, tensorflow 2.3.1+nv20.10