Specify weights or layer for dump weights callback

[szaman19]

Currently, the DumpWeights callback iterates through all the weights of a model and writes them to a specified directory.

Implementation here:

lbann/src/callbacks/dump_weights.cpp

Line 285 in 9ca712c

for (auto* w : m.get_weights()) {

For even moderately large models outputting all the weights is very time-consuming. It could be helpful for debugging purposes to have the ability to specify particular weights or layers in this callback to only dump those specific weights.

[benson31]

We should make DumpWeights, DumpGradients, and DumpErrorSignals just match the interface of DumpOutputs, which supports this feature.