# deeplearn_train_framework

**Repository Path**: ishyj/deeplearn_train_framework

## Basic Information

- **Project Name**: deeplearn_train_framework
- **Description**: 深度学习训练代码模板
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2022-12-10
- **Last Updated**: 2022-12-12

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 下载
https://gitee.com/ishyj/deeplearn_train_framework.git

# 文件及文件夹
- data：存放数据集加载器
- dataset：存放数据集
- model：存放模型定义，并需要在model/models/__init__.py中引入
- outputs：存放训练输出
- utils：工具类
- visual：可视化相关代码
- train.py：训练脚本，也可通过mode参数指定为测试脚本

# 参数
参数的引入函数位于`utils/utils.py`中的`get_args()`,其参数：
|参数|作用|类型|是否必须|选项|
|--|--|--|--|
|name|定义本次训练的代号|str|是|-|
|arch|本次训练的模型名|str|是|-|
|epochs|epoch数|int|默认200|-|
|cm|是否上传clearml|bool|默认是|-|
|mode|本次训练还是测试|str|默认训练|train、test|
|taskproject|clearml的项目名|str|||
||||||


- name
- arch
- epochs
- cm
- mode
- taskproject
- batch_size
- num_workers
- data_path
- data_type
- lr
- momentum
- weight_decay
- opti
- scheduler
- amp
- save_checkpoint
- label_smooth
- label_smooth_value
- mixup
- mixup_alpha
- Ncrop
- outputs
- best_checkpoint
- seed


# 例子
训练：
--name=UnResNet18
--arch=UnResNet18
--cm=False
--batch_size=8
--num_workers=0
--data_type=fer2013
--save_checkpoint=True
--epoch=10


测试：
--name=UnResNet18 
--arch=UnResNet18 
--cm=False 
--batch_size=8 
--num_workers=0 
--data_type=fer2013 
--save_checkpoint=True 
--epoch=10 
--mode=test 
--best_checkpoint=D:\WorkSpace\deeplearn_train_framework\outputs\UnResNet18\2022-12-10-19.39.03\checkpoints\best_checkpoint_22.0.pth