Hello,
The Python version I used to train the models is Python 3.10 and the Python on the nano board is 3.6.
For the TensorFlow package I followed this solution: Official TensorFlow for Jetson Nano! - #2 by blarish
I do not receive any unusual warnings when running the algorithm.
The only thing I changed between the two codes is the batch size, where in the GPU machine I used 32, and on the nano board I used 3. Other than that, only the Python version is different, which I am suspecting this might be the cause of the change in the performance.