Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- name: "LinearRegression"
- layer {
- name: "ip"
- type: "InnerProduct"
- bottom: "data"
- top: "ip"
- inner_product_param {
- num_output: 10
- }
- }
- layer {
- name: "softmax"
- type: "Softmax"
- bottom: "ip"
- top: "softmax"
- include {
- stage: "deploy"
- }
- }
- layer {
- name: "accuracy"
- type: "Accuracy"
- bottom: "ip"
- bottom: "label"
- top: "accuracy"
- include {
- stage: "val"
- }
- }
- layer {
- name: "loss"
- type: "SoftmaxWithLoss"
- bottom: "ip"
- bottom: "label"
- top: "loss"
- exclude {
- stage: "deploy"
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment