Dataset and models update

[isaaccorley]

• Refactored VGG loss to be more efficient
• Added DIV2K dataset (note: automatic downloading can be quite long)
• Updated models to include channels arg
• Updated examples
• Formatted with black