Required Data Science & Analytics,Data Mining & Management freelancer for Training a Model on custom dataset in pytorch (semantic Segmentation) job

I have a dataset which consist of crop images and masks. The dataloader has already build on given dataset. Now i want to train a model on that to find the accuracy and loss rate. You can use any model like Unet, Segnet or any other. But i want to train it on at least two model for comparison of results. Dataloader was build in pytorch using python. I will share some sample pictures of image and mask with shortlisted candidates.

Note: those who dont have any experience in pytorch. Please dont apply for this post.

Skills & Expertise Required

Data Science & Analytics Data Mining & Management