Retraining with imbalanced dataset

Accelerated Computing Intelligent Video Analytics TAO Toolkit
4

  1. Yes, increase the class_weight for classes with fewer samples. The enable_autoweighting cannot help for very imbalance cases.

  2. You can inspect each tfrecord file. You can refer to Tensor reshape error when evaluating a Detectnet_v2 model - #7 by Morganh