OpenTFM

项目结构

├── data
├── models
├── pretrained
├── results
├── tools
└── README.md

环境配置

# 创建虚拟环境
conda create -n OpenTFM python=3.8
conda activate OpenTFM

# 根据使用的cuda版本安装pytorch
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
pip install -r requirements.txt

# 下载数据
bash tools/download.sh

数据组织

在公开数据集外，也可以使用自定义数据集，请将数据按照如下的方式存储（TODO）

TODO

• 整体架构设计
• DataLoader
• 自定义Data测试
• ModelDriven（4/4+）
• 模型训练方法（2/2）
• 模型训练封装
• DataDriven（0/2+）
• 统一接口测试/提供baseline
• 跟驰模型测试方法（科研向）