batch_norm_fold.hpp File Reference

Folds an eval-mode BatchNorm into the Conv2D before it for the duration of an LRP explanation – Zennit's BatchNorm canonizer (roadmap FND-5, lrp_issues #8). More...

#include <vector>
#include "pulsatrix/batch_norm_module.hpp"
#include "pulsatrix/conv2d_module.hpp"

Include dependency graph for batch_norm_fold.hpp:

Go to the source code of this file.

Classes
class	pulsatrix::BatchNormFold
	While alive, merges bn's affine map into conv's weights and makes bn an exact identity; on destruction, restores both bit for bit. More...

Namespaces
namespace	pulsatrix

Detailed Description

Folds an eval-mode BatchNorm into the Conv2D before it for the duration of an LRP explanation – Zennit's BatchNorm canonizer (roadmap FND-5, lrp_issues #8).