Does it support data format X / augmentation Y / layer Z?¶
The library tries to support everything, but it could not really include everything.
The interface attempts to be flexible enough so you can put any XYZ on it. You can either implement them under the interface or simply wrap some existing Python code. See Extend Tensorpack for more details.
If you think:
The framework has limitation in its interface so your XYZ cannot be supported, OR
Your XYZ is super common / very well-defined / very useful, so it would be nice to include it.
Then it is a good time to open an issue.
How to print/dump intermediate results in training¶
How to freeze some variables in training¶
You can simply use
tf.stop_gradientin your model code in some situations (e.g. to freeze first several layers).
varreplace.freeze_variables can wrap some variables with
ScaleGradient can be used to set the gradients of some variables to 0.
Note that the above methods only prevent variables being updated by SGD.
Some variables may be updated by other means,
e.g., BatchNorm statistics are updated through the
UPDATE_OPS collection and the RunUpdateOps callback.