Skip to content

Instantly share code, notes, and snippets.

@Oktai15
Created May 7, 2018 10:00
Show Gist options
  • Save Oktai15/f2dffc09dfac22248958f82ad7c03a06 to your computer and use it in GitHub Desktop.
Save Oktai15/f2dffc09dfac22248958f82ad7c03a06 to your computer and use it in GitHub Desktop.
Caffe: simple example
name: "SimpleCaffeNet"
layer {
name: "data"
type: "Input"
top: "data"
input_param { shape: { dim: 10 dim: 1 dim: 28 dim: 28 } }
}
layer {
name: "fc1"
type: "InnerProduct"
bottom: "data"
top: "fc1"
inner_product_param {
num_output: 784
}
}
layer {
name: "relu"
type: "ReLU"
bottom: "fc1"
top: "fc1"
}
layer {
name: "fc2"
type: "InnerProduct"
bottom: "fc1"
top: "fc2"
inner_product_param {
num_output: 10
}
}
layer {
name: "prob"
type: "Softmax"
bottom: "fc2"
top: "prob"
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment