train_infer_python.txt 889 字节
Newer Older
L
lzzyzlbb 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
===========================train_params===========================
model_name:esrgan
python:python3.7
gpu_list:0
##
auto_cast:null
total_iters:lite_train_lite_infer=10
output_dir:./output/
dataset.train.batch_size:lite_train_lite_infer=100
pretrained_model:null
train_model_name:null
train_infer_img_dir:null
null:null
##
trainer:norm_train
G
gmm 已提交
16
norm_train:tools/main.py -c configs/esrgan_psnr_x4_div2k.yaml --seed 123 -o log_config.interval=5 dataset.train.num_workers=0
L
lzzyzlbb 已提交
17 18 19 20 21 22
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
W
wangna11BD 已提交
23
===========================eval_params===========================
L
lzzyzlbb 已提交
24 25 26 27 28 29 30 31
eval:null
null:null
##
===========================train_benchmark_params==========================
batch_size:32|64
fp_items:fp32
total_iters:100
--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
G
gmm 已提交
32
flags:FLAGS_cudnn_exhaustive_search=1