Updated Trainer and added some other functionalities.
Would like to merge the current status into main.
Further development will happen in issues and corresponding branches
Would like to merge the current status into main.
Further development will happen in issues and corresponding branches