Skip to content

Commit

Permalink
resnet32
Browse files Browse the repository at this point in the history
  • Loading branch information
yihui-he committed Jan 16, 2017
1 parent ff9ccba commit 5b55734
Show file tree
Hide file tree
Showing 6 changed files with 6,584 additions and 195 deletions.
2 changes: 1 addition & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ coverage.xml
*.pot

# Django stuff:
*.log
#*.log
local_settings.py

# Flask stuff:
Expand Down
10 changes: 10 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
# train resnet imagenet with caffe
### resnet-18
### resnet-32
This is a bottleneck architecture,
Since there's no strong data augmentation and 10-crop test in caffe, the results maybe a bit low.
test accuracy: accuracy@1 = 0.67892, accuracy@5 = 0.88164
training loss for resnet-32 is shown below:
the pretrained model is provided [here]()
![a](resnet_32/loss.png)

70 changes: 1 addition & 69 deletions resnet_18/resnet_18.prototxt
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ layer {
}
bias_filler {
type: "constant"
value: 0.2
value: 0
}
}
}
Expand Down Expand Up @@ -108,10 +108,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -145,10 +141,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -188,10 +180,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -231,10 +219,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -281,10 +265,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -324,10 +304,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -367,10 +343,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -417,10 +389,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -460,10 +428,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -503,10 +467,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -553,10 +513,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -590,10 +546,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -633,10 +585,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -676,10 +624,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -726,10 +670,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -769,10 +709,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down Expand Up @@ -812,10 +748,6 @@ layer {
weight_filler {
type: "msra"
}
bias_filler {
type: "constant"
value: 0.2
}
}
}
layer {
Expand Down
Loading

0 comments on commit 5b55734

Please sign in to comment.