未验证 提交 7af83680 编写于 作者: L littletomatodonkey 提交者: GitHub

fix mv3 batch size (#597)

上级 ad730ae4
...@@ -17,7 +17,7 @@ image_shape: [3, 224, 224] ...@@ -17,7 +17,7 @@ image_shape: [3, 224, 224]
LEARNING_RATE: LEARNING_RATE:
function: 'Cosine' function: 'Cosine'
params: params:
lr: 2.6 lr: 1.3
warmup_epoch: 5 warmup_epoch: 5
OPTIMIZER: OPTIMIZER:
...@@ -29,7 +29,7 @@ OPTIMIZER: ...@@ -29,7 +29,7 @@ OPTIMIZER:
factor: 0.00002 factor: 0.00002
TRAIN: TRAIN:
batch_size: 4096 batch_size: 2048
num_workers: 4 num_workers: 4
file_list: "./dataset/ILSVRC2012/train_list.txt" file_list: "./dataset/ILSVRC2012/train_list.txt"
data_dir: "./dataset/ILSVRC2012/" data_dir: "./dataset/ILSVRC2012/"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册