Skip to content
Snippets Groups Projects
Commit 7e57b20f authored by Jinhui Yuan's avatar Jinhui Yuan Committed by Will Zhang
Browse files

align alexnet to tf_benchmark (#893)

* align alexnet to tf_benchmark

* batch size 480 for 2gpu

* fix resize preprocess
parent 33a36d9b
No related branches found
No related tags found
No related merge requests found
......@@ -6,12 +6,18 @@ op {
name: "encoded"
data_type: kFloat
shape {
dim: 224
dim: 224
dim: 227
dim: 227
dim: 3
}
encode_case {
jpeg {
preprocess {
resize {
width: 227
height: 227
}
}
}
}
}
......@@ -45,7 +51,7 @@ op {
in: "transpose/out"
out: "out"
filters: 64
padding: "SAME"
padding: "VALID"
data_format: "channels_first"
kernel_size: 11
kernel_size: 11
......@@ -63,23 +69,23 @@ op {
}
}
op {
name: "lrn1"
local_response_normalization_conf {
in: "conv1/out"
out: "out"
depth_radius: 2
bias: 2
alpha: 0.0001
beta: 0.75
data_format: "channels_first"
}
}
#op {
# name: "lrn1"
# local_response_normalization_conf {
# in: "conv1/out"
# out: "out"
# depth_radius: 2
# bias: 2
# alpha: 0.0001
# beta: 0.75
# data_format: "channels_first"
# }
#}
op {
name: "pool1"
max_pooling_2d_conf {
in: "lrn1/out"
in: "conv1/out"
out: "out"
padding: "VALID"
data_format: "channels_first"
......@@ -114,23 +120,23 @@ op {
}
}
op {
name: "lrn2"
local_response_normalization_conf {
in: "conv2/out"
out: "out"
depth_radius: 2
bias: 2
alpha: 0.0001
beta: 0.75
data_format: "channels_first"
}
}
#op {
# name: "lrn2"
# local_response_normalization_conf {
# in: "conv2/out"
# out: "out"
# depth_radius: 2
# bias: 2
# alpha: 0.0001
# beta: 0.75
# data_format: "channels_first"
# }
#}
op {
name: "pool2"
max_pooling_2d_conf {
in: "lrn2/out"
in: "conv2/out"
out: "out"
padding: "VALID"
data_format: "channels_first"
......
......@@ -4,7 +4,7 @@ globalfs_conf {
}
}
data_part_num: 30
piece_size: 240
piece_size: 480
model_load_snapshot_path: ""
max_data_id_length: 0
train_conf {
......@@ -18,12 +18,11 @@ train_conf {
std: 0.01
}
}
batch_size: 240
batch_size: 480
l2: 0.0005
model_update_conf {
learning_rate: 0.01
momentum_conf {
beta: 0.9
naive_conf {
}
learning_rate_decay {
exponential_conf {
......
placement_group {
op_set {
op_name: "decode"
op_name: "transpose"
}
parallel_conf {
policy: kDataParallel
......@@ -10,12 +11,11 @@ placement_group {
placement_group {
op_set {
op_name: "transpose"
op_name: "conv1"
op_name: "lrn1"
# op_name: "lrn1"
op_name: "pool1"
op_name: "conv2"
op_name: "lrn2"
# op_name: "lrn2"
op_name: "pool2"
op_name: "conv3"
op_name: "conv4"
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment