Distracted driver detection with Jetson Nano

Please review my pet project for distracted driver detection. I have used the Resnet model for it.
Code: https://github.com/gasatig/Jetson/blob/master/driver_resnet.ipynb
Pre-trained Model: https://drive.google.com/file/d/196j5f_1JF63ZUMLkIRItKyXs_t1N_0Io/view?usp=sharing
Data Source: https://www.kaggle.com/c/state-farm-distracted-driver-detection
Please check the last cell for accuracy and cross-entropy for the model.