Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- name: "iris"
- layer {
- name: "data"
- type: "HDF5Data"
- top: "data"
- top: "label"
- include: {
- phase: TRAIN
- }
- hdf5_data_param {
- source: "./files/train_file_location.txt"
- batch_size: 10
- }
- }
- layer {
- name: "data"
- type: "HDF5Data"
- top: "data"
- top: "label"
- include: {
- phase: TEST
- }
- hdf5_data_param {
- source: "./files/test_file_location.txt"
- batch_size: 10
- }
- }
- layer {
- name: "ip1"
- type: "InnerProduct"
- bottom: "data"
- top: "ip1"
- inner_product_param {
- num_output: 20
- weight_filler {
- type: "xavier"
- }
- bias_filler {
- type: "constant"
- }
- }
- }
- layer {
- name: "relu1"
- type: "ReLU"
- bottom: "ip1"
- top: "ip1"
- }
- layer {
- name: "ip2"
- type: "InnerProduct"
- bottom: "ip1"
- top: "ip2"
- inner_product_param {
- num_output: 3
- weight_filler {
- type: "xavier"
- }
- bias_filler {
- type: "constant"
- }
- }
- }
- layer {
- name: "softmax"
- type: "Softmax"
- bottom: "ip2"
- top: "smip2"
- }
- layer {
- name: "loss"
- type: "SoftmaxWithLoss"
- bottom: "ip2"
- bottom: "label"
- top: "loss"
- }
- layer {
- name: "accuracy"
- type: "Accuracy"
- bottom: "smip2"
- bottom: "label"
- top: "accuracy"
- include {
- phase: TEST
- }
- }
Add Comment
Please, Sign In to add comment