Add mention of issue FluxML#2040

christiangnrd · Aug 30, 2022 · f5793b5 · f5793b5
1 parent bb18173
commit f5793b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/functor.jl b/src/functor.jl
@@ -88,7 +88,7 @@ function params(m...)
   return ps
 end
 
-# Allows caching of the parameters when params is called within gradient()
+# Allows caching of the parameters when params is called within gradient() to fix #2040.
 @non_differentiable params(m...)
 
 struct FluxCUDAAdaptor end