Modify ML Training Script

I am working on a project to use deep learning techniques for game automation. I am working with the game Dirt 5 (racing game) and need the scripts attached modified to train data I have. I am working with a CNN, so I need techniques like Resnet, VGG, Xception. These are also light weight which is what is needed.