Я пытаюсь обучить CNN с одним выходом с двумя наборами данных. Я хочу, чтобы в каждом пакете можно было получать некоторые данные из первого набора данных и некоторые данные из второго набора данных, и каждый набор данных имеет свою соответствующую функцию потерь. Я читал статьи, где, если у нейронной сети есть две головы, это возможно, но я не уверен, как сделать два набора данных с двумя потерями в одной нейронной сети с одним выходом. Я также был бы открыт для использования pytorch, если есть некоторые существующие примеры.