PaddleX 3d任务模型配置文件参数说明¶
Global¶
| 参数名 | 数据类型 | 描述 | 默认值 |
|---|---|---|---|
| model | str | 指定模型名称 | yaml文件中指定的模型名称 |
| mode | str | 指定模式（check_dataset/train/evaluate/export/predict） | check_dataset |
| dataset_dir | str | 数据集路径 | yaml文件中指定的数据集路径 |
| device | str | 指定使用的设备 | yaml文件中指定的设备id |
| output | str | 输出路径 | "output" |
| load_cam_from | str | cam分支的预训练参数路径 | yaml文件中指定的cam分支的预训练参数路径 |
| load_lidar_from | str | lidar分支的预训练参数路径 | yaml文件中指定的lidar分支的预训练参数路径 |
| datart_prefix | bool | 数据集路径是否需要加上根路径 | True |
| version | str | 数据集版本号 | "mini" |
CheckDataset¶
| 参数名 | 数据类型 | 描述 | 默认值 |
|---|---|---|---|
| convert.enable | bool | 是否进行数据集格式转换 | False (当前不支持) |
| split.enable | bool | 是否重新划分数据集 | False (当前不支持) |
Train¶
3d任务公共参数¶
| 参数名 | 数据类型 | 描述 | 默认值 |
|---|---|---|---|
| epochs_iters | int | 模型对训练数据的重复学习次数 | yaml文件中指定的重复学习次数 |
| batch_size | int | 批大小 | yaml文件中指定的批大小 |
| learning_rate | float | 初始学习率 | yaml文件中指定的初始学习率 |
| warmup_steps | str | 训练模型的开始阶段，进行热身训练的步数 | yaml文件中指定的热身训练的步数 |
Evaluate¶
| 参数名 | 数据类型 | 描述 | 默认值 |
|---|---|---|---|
| batch_size | int | 批大小 | yaml文件中指定的批大小 |
| weight_path | str | 评估模型路径 | 默认训练产出的本地路径，当指定为None时，表示使用官方权重 |
Export¶
| 参数名 | 数据类型 | 描述 | 默认值 |
|---|---|---|---|
| weight_path | str | 导出模型的动态图权重路径 | 默认训练产出的本地路径，当指定为None时，表示使用官方权重 |
Predict¶
| 参数名 | 数据类型 | 描述 | 默认值 |
|---|---|---|---|
| batch_size | int | 预测批大小 | yaml文件中指定的预测批大小 |
| model_dir | str | 预测模型路径 | 默认训练产出的本地推理模型路径，当指定为None时，表示使用官方权重 |
| input | str | 预测输入路径 | yaml文件中指定的预测输入路径 |
| kernel_option.run_mode | str | 推理引擎设置，如: "paddle" | paddle |