@Mahmoud_Abdelkhalek Ahh yes, that make sense. I get you now. That is great, I can just step into the model up till the forward() method, and then track from there. Thanks so much. Haha, that was easier than I thought. I was probably overthinking it, figuring that it would be harder than that. But what you suggested is easy enough. Thanks again :).