提供针对性的配置文件
时间：2022-03-07 00:54:01 | 来源：行业动态
时间：2022-03-07 00:54:01 来源：行业动态
：要实现指令的单纯化的目标，就需要有完整的配置系统作为搭配，这个部分会是开发人员在TAO工具上最需要花时间的内容。
由于每种神经网络都有其独特的结构与参数种类，因此必须为每个网络提供各自的配置文件，例如图像分类(classification)的配置文件中，都有model_config、train_config、eval_config三个配置组，物件检测(object detection)配置文件还多了nms_config、augmentation_config、dataset_config等配置组。
此外，在训练过程还有dataset_convert、train、prune、retrain等阶段，都有一小部分参数需要进行微调，这些细节就需要开发人员对个别神经网络结构有进一步的了解，所幸TAO为目前所支持的神经网络都提供优化过的配置文件，使用者不需要从0开始进行配置，只要以范例文件为基础进行修改就可以。
绝大部分时候，我们需要修改的部分，就只有一些文件路径(xxx_path)、训练回合数(num_epoch)、批量数(batch_size_per_gpu)，以及dataset_config里面的类别名称(target_class_mapping)的对应值。除非有特别的要求，并且您熟悉该神经网络的结构与特性，否则使用英伟达提供的优化配置参数，是足以获得一定水平之上的效果。
-
在线咨询
