Максимально всрато вышло, хз почему. Нужно датасет подгонять под флюкс, и видимо заменить теги на caption в обучении.