Multiple new features for training

From History:

• New Network: Light-UNet
• Misc bugfixes
• Automatic check-pointing of the model has been added
• Extending the codebase has been simplified
• New optimizers added
• New metrics added
• Affine augmentation can now be significantly fine-tuned
• Update logic for penalty calculation
• RGB-specific augmentation added
• Cropping added