mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 07:01:39 +08:00
45 lines
1.7 KiB
Plaintext
45 lines
1.7 KiB
Plaintext
=========================== base_train ===========================
|
|
model_name:ResNet50
|
|
python:python3.7
|
|
gpu_list:0
|
|
-o Global.device:gpu
|
|
-o Global.auto_cast:null
|
|
-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
|
|
-o Global.output_dir:./output/
|
|
-o DataLoader.Train.sampler.batch_size:8
|
|
-o Global.pretrained_model:null
|
|
train_model_name:latest
|
|
train_infer_img_dir:./dataset/ILSVRC2012/val
|
|
null:null
|
|
##
|
|
trainer:norm_train
|
|
norm_train:tools/train.py -c ppcls/configs/ImageNet/ResNet/ResNet50.yaml -o Global.seed=1234 -o DataLoader.Train.sampler.shuffle=False -o DataLoader.Train.loader.num_workers=0 -o DataLoader.Train.loader.use_shared_memory=False -o Global.eval_during_train=False -o Global.save_interval=2 -o Global.print_batch_step=1
|
|
pact_train:null
|
|
fpgm_train:null
|
|
distill_train:null
|
|
to_static_train:-o Global.to_static=True
|
|
null:null
|
|
##
|
|
=========================== amp_train ===========================
|
|
model_name:ResNet50
|
|
python:python3.7
|
|
gpu_list:0
|
|
-o Global.device:cpu
|
|
-o Global.auto_cast:null
|
|
-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
|
|
-o Global.output_dir:./output/
|
|
-o DataLoader.Train.sampler.batch_size:8
|
|
-o Global.pretrained_model:null
|
|
train_model_name:latest
|
|
train_infer_img_dir:./dataset/ILSVRC2012/val
|
|
null:null
|
|
##
|
|
trainer:amp_train
|
|
amp_train:tools/train.py -c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml -o Global.seed=1234 -o DataLoader.Train.sampler.shuffle=False -o DataLoader.Train.loader.num_workers=0 -o DataLoader.Train.loader.use_shared_memory=False -o AMP.scale_loss=128 -o AMP.use_dynamic_loss_scaling=True -o AMP.level=O1 -o Global.print_batch_step=1 -o Global.use_dali=False -o Global.eval_during_train=False
|
|
pact_train:null
|
|
fpgm_train:null
|
|
distill_train:null
|
|
to_static_train:-o Global.to_static=True
|
|
null:null
|
|
##
|