Update readme

Add model.eval()

Update for PyTorch 1.7 and PyTorch Lightning 1.1

• Update for new a version of PyTorch and PyTorch Lightning
• Reduce VGG model size
• Provide detail WandB report

Retrain all models to have better accuracy

Using cosine annealing improve all models' accuracy about 1% to 2%

Switch to PyTorch Lightning

• Code is now re-producible
• Simplier interface

Better handling relative paths

1.2

Simplify training code

Stable

Work pretty well. Has enough models for general use.