multi task训练torch_GitHub - xrick/Pytorch-Multi-Task-Multi-class-Classification: 旨在搭建一个分类问题在Pytorch框架...

1. Pytorch-Multi-Task-Multi-class-Classification

MTMC-Pytorch:

MTMC-Pytorch = Multi-Task Multi-Class Classification Project using Pytorch.

目的：

旨在搭建一个分类问题在Pytorch框架下的通解，批量解决单任务多分类问题、多任务多分类问题。

使用：

需要做的准备工作是将样本整理成如下格式按文件夹存放；

MTMC自动解析任务获取类别标签、自适应样本均衡、模型训练、模型评估等过程。

MLDataloader load MTMC dataset as following directory tree.

Make sur train-val directory tree keeps consistency.

data_root_path

├── task_A

│ ├── train

│ │ ├── class_1

│ │ ├── class_2

│ │ ├── class_3

│ │ └── class_4

│ └── val

│ ├── class_1

│ ├── class_2

│ ├── class_3

│ └── class_4

└── task_B

├── train

│ ├── class_1

│ ├── class_2

│ └── class_3

└── val

├── class_1

├── class_2

└── class_3

备注&#x