Update README.md

RaoUmer · Mar 24, 2021 · b099918 · b099918
1 parent 8f11db8
commit b099918
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -35,6 +35,9 @@ This model can be run on arbitrary images with a Docker image hosted on Replicat
 - [PyTorch >= 1.0](https://pytorch.org/) (CUDA version >= 8.0 if installing with CUDA.)
 - Python packages:  `pip install numpy opencv-python`
 
+#### Train models
+- The SR training code is based on the [SRResCGAN](https://github.com/RaoUmer/SRResCGAN/tree/master/training_codes).
+
 #### Test models
 1. Clone this github repository as the following commands:
 ```