diff --git a/README.md b/README.md index 9f94a74..392ef1d 100644 --- a/README.md +++ b/README.md @@ -16,3 +16,15 @@ conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch ##### CPU Only conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch + + +#### Train your Model: +!python main.py --LR_path custom_dataset_cars/hr_train_LR --GT_path custom_dataset_cars/hr_train_HR + +#### Test your Model: +!python main.py --mode test_only --LR_path test_data/cars --generator_path ./model/srgan_custom.pt + + +#### If you want to test the model which is trained on div2K dataset then download the model from here: https://github.com/AarohiSingla/SRGAN_CustomDataset/tree/main/pretrained_models + +