fix #622: replaced array_type with atype

denizyuret · Oct 22, 2020 · db3f2db · db3f2db
1 parent 8a4fdbf
commit db3f2db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/train20/train_ka.jl b/src/train20/train_ka.jl
@@ -1,4 +1,5 @@
+using Knet: atype
 using Knet.KnetArrays: KnetArray
-param(x::KnetArray; atype=array_type[]) = Param(convert(atype,x))
+param(x::KnetArray; atype=atype()) = Param(convert(atype,x))
 update!(w::KnetArray, g, p) = gclip_update!(w, g, p)
 _optimizers(::KnetArray,otype; o...) = otype(;o...)