# The train/test net protocol buffer definition net: "./train.prototxt" # test_iter specifies how many forward passes the test should carry out. # In our case, 311*64=19904 is above the total number of validation images on CelebA (19867) test_iter: 311 # Carry out testing every 1000 training iterations (same as display). test_interval: 1000 # The base learning rate, momentum and the weight decay of the network. base_lr: 1e-05 momentum: 0.0 weight_decay: 1e-05 # The learning rate policy lr_policy: "inv" gamma: 0.0001 power: 0.75 # Display every 1000 iterations display: 1000 # The maximum number of iterations max_iter: 100000 # snapshot intermediate results snapshot: 5000 snapshot_prefix: "./train_snapshots" # solver mode: CPU or GPU solver_mode: GPU type: "RMSProp" rms_decay: 0.98