1)  data_set.py  -  processing img dataset
2)  model_trainig.py  -  creating and training a model and saving it as a file "model_my01.pth"
3)  main.py  -  testing of the pytorch and tensorrt model  

Result:
fps:  PyTorch: 150

accuracy: PyTorch:83%

Dataset, trained model:https://drive.google.com/file/d/1jePR2Wef7eSGknFcWGnFdYJjP8zBAeqB/view?usp=sharing