TAO Classification - Unbalanced Dataset

[matt.berseth]

I am training a classification network (efficientnet_b0) on a dataset where I have unbalanced images.

Does each minibatch have equal number of examples from each class of data? Are there any config options for how the data is loaded or how sampling is handled?

• Hardware: V100
• Network Type: Classification

[Morganh]

@matt.berseth

Does each minibatch have equal number of examples from each class of data?

No. It is not needed.

@matt.berseth

Are there any config options for how the data is loaded or how sampling is handled?

There is not parameters in the spec file.

[matt.berseth]

@Morganh

No. It is not needed.

Thanks Morganh - any links or references on why minibatches aren’t balanced? Is some other technique used (weights or something)?

Thanks for the info on not having data set options via the spec file.

[Morganh]

Sorry, each minibatch has equal number of examples from each class of data.
I misunderstood your question. What I mentioned earlier is that for training images, it is not needed to set equal number of examples for each class.