Switch to unified view

a b/train_densenet_albumentations.ipynb