What is the module for classification model.

[AhmadShaik]

I am trying to train a classification model. I would like to know what to pass as an argument for module in the below configuration.
python train_tlt.py --module detectnet_v2 -m nvidia/tlt_pretrained_detectnet_v2:resnet18
--arch detectnet_v2 --dataset-task
--dataset-export-spec example_specs/dataset_export_spec.txt -c specs/
--key --model_name model --experiment_spec_file specs/detectnet_v2_spec_file_template.txt