Most vai_p_caffe tasks require a configuration file as an input
argument. A typical configuration file is shown below:
workspace: "examples/decent_p/"
gpu: "0,1,2,3"
test_iter: 100
acc_name: "top-1"
model: "examples/decent_p/float.prototxt"
weights: "examples/decent_p/float.caffemodel"
solver: "examples/decent_p/solver.prototxt"
rate: 0.1
pruner {
method: REGULAR
}The definition for the terms used are:
- workspace
- Directory for saving temporary and output files.
- gpu
- Use the given GPU devices IDS separated by ',' for acceleration.
- test_iter
- The number of iterations to use in test phase. A larger value improves the analysis results but increases run time. The maximum value of this parameter is determined by the size of the validation dataset/batch_size, i.e, all data in the validation dataset will be used for testing.
- acc_name
- The accuracy measure to be used to determine the "goodness" of the model.
- model
- The model definition protocol buffer text file. If there are two separate model definition files used in training and testing, merge them into a single file.
- weights
- The model weights to be pruned.
- solver
- The solver definition protocol buffer text file used for finetuning.
- rate
- The weight reduction parameter sets the amount by which the number of computations will be reduced relative to the baseline model. For example, with a setting of "0.1," the tool would attempt to reduce the number of multiply-add operations by 10% relative to the baseline model.
- method
- Pruning method is used. Currently REGULAR is the only valid value.