Я хочу использовать MixUp в качестве дополнения к данным для задачи классификации изображений с несколькими метками, но кажется, что он поддерживает только метки с формой (batch_size,). Как мне изменить это, чтобы разрешить использование таких меток, как [0.0, 1.0, 0.0, 1.0]
Я пробовал вызвать argmax для моего вектора меток [0.0, 1.0] , 0.0, 1.0], что дает мне один индекс. Это заставляет функцию работать, но она смешивает только один класс.
Подробнее здесь: https://stackoverflow.com/questions/784 ... el-problem