I have a weighted dataset, and a model composed of two parts.
How can I train the model in such a way that the dataset weights only apply to the first part of it (whi