ppo_net.txt

name: "PPONet"

layer { 
  name: 'input' 
  type: 'Input' 
  top: "input"
  top: "label"
  top: "weights"
  top: "param"
  input_param {
	shape: { dim: 256 dim: 1 dim: 1 dim: 5 } 
  shape: { dim: 256 dim: 1 dim: 1 dim: 2 }
  shape: { dim: 256 dim: 1 dim: 1 dim: 1 }
  shape: { dim: 1 dim: 1 dim: 1 dim: 4 }
  }

  include: { phase: TRAIN }
}

layer { 
  name: 'input'
  type: 'Input'
  top: 'input'
  input_param { 
    shape: { dim: 1 dim: 1 dim: 1 dim: 5 } 
  } 

  include: { phase: TEST }
}

layer {
  name: "ip0"
  type: "InnerProduct"
  bottom: "input"
  top: "ip0"
  param {
    lr_mult: 1
	decay_mult: 1
  }
  param {
    lr_mult: 2
	decay_mult: 0
  }
  inner_product_param {
    num_output: 256
    weight_filler {
      type: "xavier"
    }
    bias_filler {
      type: "constant"
    }
  }
}

layer {
  name: "relu0"
  type: "ReLU"
  bottom: "ip0"
  top: "relu0"
}

layer {
  name: "ip1"
  type: "InnerProduct"
  bottom: "relu0"
  top: "ip1"
  param {
    lr_mult: 1
	decay_mult: 1
  }
  param {
    lr_mult: 2
	decay_mult: 0
  }
  inner_product_param {
    num_output: 128
    weight_filler {
      type: "xavier"
    }
    bias_filler {
      type: "constant"
    }
  }
}

layer {
  name: "relu1"
  type: "ReLU"
  bottom: "ip1"
  top: "relu1"
}

layer {
  name: "output"
  type: "InnerProduct"
  bottom: "relu1"
  top: "output"
  param {
    lr_mult: 1
	decay_mult: 1
  }
  param {
    lr_mult: 2
	decay_mult: 0
  }
  inner_product_param {
    num_output: 2
    weight_filler {
      type: "xavier"
    }
    bias_filler {
      type: "constant"
    }
  }
}

layer {
  name: "loss"
  type: "PPOLoss"
  bottom: "output"
  bottom: "label"
  bottom: "weights"
  bottom: "param"
  top: "loss"

  include: { phase: TRAIN }
}