Add Swinir modules ( SwinIR torch implementation and Lightning module)

mlangguth89 / downscaling_benchmark

6 stars 0 forks source link

Open ankitpatnala opened 6 months ago

ankitpatnala commented 6 months ago

Things to do :

Copy the original implementation from SwinIR : github_repo
Create a PyTorch lightning module of SwinIR taking args similar to the model Class defined in model_utils.py