Add reference

basicsr/archs/focalir_arch.py

@@ -1,10 +1,20 @@
+######
+# FocalIR
+# This code is referenced by Focal Transformer and SwinIR
+# This model is supported by BasicSR
+######
 # --------------------------------------------------------
 # Focal Transformer
 # Copyright (c) 2021 Microsoft
 # Licensed under The MIT License [see LICENSE for details]
 # Written by Jianwei Yang (jianwyan@microsoft.com)
 # Based on Swin Transformer written by Zhe Liu
 # --------------------------------------------------------
+# -----------------------------------------------------------------------------------
+# SwinIR: Image Restoration Using Swin Transformer, https://arxiv.org/abs/2108.10257
+# Originally Written by Ze Liu, Modified by Jingyun Liang.
+# -----------------------------------------------------------------------------------
+
 
 import math
 import torch