From 8635f895da6ca6d5ba42e18059cbcda8df0e33dd Mon Sep 17 00:00:00 2001
From: Chen Liangyu <chenliangyux@gmail.com>
Date: Tue, 12 Apr 2022 11:32:49 +0800
Subject: [PATCH] Update NAFNet_arch.py

---
 basicsr/models/archs/NAFNet_arch.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/basicsr/models/archs/NAFNet_arch.py b/basicsr/models/archs/NAFNet_arch.py
index 638a0bd..119786b 100644
--- a/basicsr/models/archs/NAFNet_arch.py
+++ b/basicsr/models/archs/NAFNet_arch.py
@@ -5,6 +5,12 @@
 '''
 Simple Baselines for Image Restoration
 
+@article{chen2022simple,
+  title={Simple Baselines for Image Restoration},
+  author={Chen, Liangyu and Chu, Xiaojie and Zhang, Xiangyu and Sun, Jian},
+  journal={arXiv preprint arXiv:2204.04676},
+  year={2022}
+}
 '''
 
 import torch