Bda 1216 improve logging

[durtschi]

Added several additions to logging primarily for wandb. Also added some additional input options for better logging clearity.

Look at wandb run "exome_test_211201" for example output

There is some redundancy between wandb info and config sections but it's useful to have values in the config section for grouping in reports, etc.

Git Repo

• Git branch and commit info in wandb run info section
• Wandb gives a copy and paste git command to make a local branch with the correct commit for the run
• Git info including most recent commit message in wandb config section

Run Name and Notes

• command line input run name
• command line input run notes (a more extensive message about the run that shows up right below the run name kind of like a git commit message)

Timing Data (shown in plots)

• avg batch time per epoch
• avg fractions of batch time for many training steps: decompress, load, train total, forward pass, backward pass, optimize, etc.