Error in Using CutMixGenerator

[younkyul]

When I run a simple CNN with CutMixgenerator,

train_gen = CutMixImageDataGenerator(
generator1 = train_generator1,
generator2 = train_generator2,
img_size = image_size,
batch_size = 48,
)
...
model.add(Flatten())
model.add(Dense(2))
model.add(Activation('softmax'))
model.compile(loss='binary_crossentropy', optimizer='adadelta')
model.fit_generator(generator = train_gen, epochs=10, steps_per_epoch=100)

I got an error as follows:

expected activation_5 to have shape (2,) but got array with shape (48,)

Please let me know how to resolve it

[deepakcrk]

May be its due to you are using binary_crossentropy, if you are using categorical_crossentropy this may be resolved.