An error occurred while training with TLT

Please refer to ZeroDivisionError when training peoplenet - #10 by Morganh

Your val images is only 4. It is smaller than num_shards.
Please add more val images or set smaller num_shards.

val_images is (val_split)% of total images.train_images is (100-val_split)% of total images.

Please make sure below at the same time.

  1. val_images >= num_shards
  2. train_images >= num_shards