Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
root
12d91d88a5
|
4 months ago | |
|---|---|---|
| deepsparkhub | 4 months ago | |
| README.md | 4 months ago | |
README.md
整体流程
- 获取模型库代码和文档
cd /usr
git clone -b openi-task12 https://gitee.com/deep-spark/deepsparkhub
- 按照每个模型各自的readme文档，进行训练
在/usr目录下进行git操作！！/code目录下无法git
resnet50
- 进入目标文件夹
cd /usr/deepsparkhub/cv/classification/resnet50/pytorch/
- 修改train_resnet50_imagenet.sh 中的超参数
vim ./scripts/train_resnet50_imagenet.sh
- 解压数据集
# 拷贝到/usr目录
cp /dataset/imagenet.zip /usr
# 解压
unzip /usr/imagenet.zip
- 执行训练
bash scripts/fp32_1card.sh --data-path /usr/imagenet
googlenet
-
修改train.py
-
执行训练
python3 train.py --data-path /usr/imagenet --model googlenet --batch-size 512
VGG16
- 安装依赖
pip3 install -r requirements.txt
- 修改配置文件
vim train_vgg16_amp_dist.sh
# 修改数据集路径
DATA_PATH=${DATA_PATH:-"/usr/imagenet"}
# 修改 epochs
--epochs 5
- 执行训练
bash train_vgg16_amp_dist.sh
lenet
- 修改超参数
vim train.py
- 执行训练
python3 train.py --data-path /usr/imagenet --model lenet
alexnet
- 修改超参数
vim train.py
- 执行训练
cd start_scripts
bash train_alexnet_torch.sh --data-path /usr/imagenet
swin_transfer
- 安装依赖
pip install timm==0.4.12
pip install yacs
- 修改超参数
vim config.py
- 执行脚本
python3 -m torch.distributed.launch --nproc_per_node 1 --master_port 12345 main.py \
--cfg configs/swin/swin_tiny_patch4_window7_224.yaml --data-path /usr/imagenet --batch-size 128
mobilenetv3
- 修改配置文件
vim train_mobilenet_v3_large_amp_dist.sh
# 修改数据集路径
DATA_PATH=${DATA_PATH:-"/usr/imagenet"}
# 修改 epoch
--epochs 5
- 运行脚本
bash train_mobilenet_v3_large_amp_dist.sh
官方文档有误（数据集路径，文件名）
InceptionV3(同上)
- 修改配置文件
vim train_inception_v3_amp_dist.sh
# 修改数据集路径
DATA_PATH=${DATA_PATH:-"/usr/imagenet"}
# 修改 epoch
--epochs 5
- 运行脚本
bash train_inception_v3_amp_dist.sh
官方文档有误（数据集路径）
densenet121
-
修改train.py
-
执行训练
python3 train.py --data-path /usr/imagenet --model densenet201 --batch-size 128
第二周
mobilenetv2
dpn107
resnest101
resnest14
wide_resnet101
ResNeXt50_32x4d
ResNeXt101_32x8d
SEResNeXt
squeezenet
python3 train.py --data-path /usr/imagenet --model squeezenet1_0 --lr 0.001 --epochs 5
第三周
acmix
cbam
cspdarknet53
dpn92
eca_resnet152
eca_mobilenetv2
efficent_b4
InceptionV4
xception
No Description
Python Unity3D Asset Text Jupyter Notebook C++ other
Contributors (15)
root@ta49c1bef89241678c28a09ff2ab8516-task0-0.ta49c1bef89241678c28a09ff2ab8516.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@va0ce927ad584c97b54d467d196e43ac-task0-0.va0ce927ad584c97b54d467d196e43ac.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@f65200ce1e834185a6c69cc56c333fad-task0-0.f65200ce1e834185a6c69cc56c333fad.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@g5958d6d358349baae3f94d4f2af4e84-task0-0.g5958d6d358349baae3f94d4f2af4e84.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@k8c597bf075f422fb6dd10334526f457-task0-0.k8c597bf075f422fb6dd10334526f457.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@qda09efbcf0c43058da3274d380d3575-task0-0.qda09efbcf0c43058da3274d380d3575.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@qeadd265855c47edadcba634969ecb3f-task0-0.qeadd265855c47edadcba634969ecb3f.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@c8b5df28cadf4a96ae6fb064ae2ff57f-task0-0.c8b5df28cadf4a96ae6fb064ae2ff57f.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@cc3ca68f82494cd4951ec879b097603e-task0-0.cc3ca68f82494cd4951ec879b097603e.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@iea39d820f9c43db88b230fc7d98c81f-task0-0.iea39d820f9c43db88b230fc7d98c81f.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@mf903b0ec0b04870a3f8098a2ea47511-task0-0.mf903b0ec0b04870a3f8098a2ea47511.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@nf3c692931e24ebf843649d080400174-task0-0.nf3c692931e24ebf843649d080400174.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@o554b9d53e69438ba86b2c0175b709ec-task0-0.o554b9d53e69438ba86b2c0175b709ec.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local
root@w83d9fbd96604f109df7c4011ca838b6-task0-0.w83d9fbd96604f109df7c4011ca838b6.c1a35ae42c044fd6b2a2be3b7a534094.svc.cluster.local