From f3c129ef86291d9738536a62b2c11479b3e09c5f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?BU=20Fanchen=20=E5=8D=9C=E5=87=A1=E8=BE=B0?=
 <boqvezen97@kaist.ac.kr>
Date: Wed, 30 Aug 2023 09:41:30 +0900
Subject: [PATCH] added reference for AttentionModel

---
 rl4co/models/zoo/am/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rl4co/models/zoo/am/model.py b/rl4co/models/zoo/am/model.py
index 685e7206..10b1dfb9 100644
--- a/rl4co/models/zoo/am/model.py
+++ b/rl4co/models/zoo/am/model.py
@@ -7,7 +7,7 @@
 
 
 class AttentionModel(REINFORCE):
-    """Attention Model based on REINFORCE.
+    """Attention Model based on REINFORCE: https://arxiv.org/abs/1803.08475.
 
     Args:
         env: Environment to use for the algorithm