Saving custom layer parameters

It plots directly to your tensorboard logs.