How to change the architecture of the network in running time

14 views

caffephaseprototxt

Skip to first unread message

kiana ehsani

unread,

Jul 27, 2016, 1:37:34 PM7/27/16

to Caffe Users

I want to add/remove some layers in the running time. For example, I have a variable called A. If A is zero I want to have three layers and if A is not zero I want to have 5 layers in my architecture. These layers are ROI pooling layers (I don't know whether it helps or not). I know that adding and removing layers in the prototxt file is really easy but I have no idea how to do this in the running time. Even having something like

include {
    phase: TRAIN
}

with 4-5 different phases will work. I am just familiar with TRAIN and TEST but I want something like TRAIN1, TRAIN2, TRAIN3,....