23 lines
905 B
Plaintext
23 lines
905 B
Plaintext
=========================== amp_train ===========================
|
|
model_name:ResNet152
|
|
python:python3.7
|
|
gpu_list:0|0,1
|
|
-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:amp_train
|
|
amp_train:tools/train.py -c ppcls/configs/ImageNet/ResNet/ResNet152.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=O2 -o Global.eval_during_train=False -o Global.save_interval=2
|
|
pact_train:null
|
|
fpgm_train:null
|
|
distill_train:null
|
|
to_static_train:-o Global.to_static=True
|
|
null:null
|
|
##
|