Save recon_loss and KL_loss separately to visualize tradeoff

[gwaybio]

Visualizing how each loss function contributes to the overall loss and how each is learned through each training epoch will be important. It will be important especially if warmup is implemented in #12

[gwaybio]

Possibly of interest will be generating a custom metric for model compilation