Please provide the following information when requesting support.
• Hardware (T4/V100/Xavier/Nano/etc)
• Network Type (Detectnet_v2/Faster_rcnn/Yolo_v4/LPRnet/Mask_rcnn/Classification/etc)
• TLT Version (Please run “tlt info --verbose” and share “docker_tag” here)
• Training spec file(If have, please share here)
• How to reproduce the issue ? (This is for errors. Please share the command line and the detailed log here.)
i tried to train yolov4 model as followed with the instructions of tao toolkit, however, when i made the config of class_weighting_config in training spec .*txt format, the spec zip file is downloaded from the website of toolkit as well, the training log popped the errors like :
Message type “YOLOv3DatasetConfig” has no field named “class_weighting_config”, it is so weird, i am sure that i referred to the correct spec file which is located on the yolov4 folder…
here is i read from website
Class Weighting Config
YOLOV4 supports class-level weighting on the loss function during training. The following is an example of the class weighting configuration (class_weighting_config) to set weights for two classes.
Class Weighting Config
YOLOV4 supports class-level weighting on the loss function during training. The following is an example of the class weighting configuration (class_weighting_config) to set weights for two classes.
class_weighting_config {
class_weighting{
key: “person”
value: 1.0
}
class_weighting{
key: “bus”
value: 5.0
}
}
The parameters in class_weighting_config are defined as follows:
class_weighting: This parameter maps the class name to the corresponding class weight on the loss function. The class weight value should be greater than 0. If a class weight is not explictly set in the config, a default value of 1.0 will be assigned implictly.