Train and evaluate task specific models performance on the specific tasks on GPU

[afmjoaa]

• [ ] Make the task-specific model GPU trainable.
• [ ] Train model for the selected 3 tasks and save them.
• [ ] Evaluate the performance of the model using the selected metrics.