Missing references

[MikePelton]

Hi - hardly an issue, more some observations - you need to add kornia, lpips and tensorboard to your list of dependencies to run train.py. Also the version of PyTorch you install needs to play nicely with the installed local version of the CUDA Toolkit - if the installation fails you're pointed to a website with appropriate instructions so no need for those to be reproduced here. Thanks for sharing this fantastic work!

[apchenstu]

Oh, thanks! I will update the Installation in the next version~