Merge pull request #88 from bokveizen/patch-2

added reference for AttentionModel
ai4co · Aug 30, 2023 · c00c5c1 · c00c5c1
2 parents aa7bd31 + f3c129e
commit c00c5c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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