Thanks for replying. Here is the code 6.
This is not a small code, however, all the starting cells are just loading the required files and the code.
The last cell is where I run the train.py function
Please let me know if more clarification is required.