Merge pull request #314 from FluxML/adam

Replace ADAM as Adam

Author: yuehhua

examples/digitsum_deepsets.jl

@@ -117,8 +117,8 @@ function train(; kws...)
     ρ = Dense(args.hidden_dims[3], args.target_dim)
     model = DeepSet(ϕ, ρ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/gae.jl

@@ -74,8 +74,8 @@ function train(; kws...)
 
     model = GAE(encoder, σ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/gat.jl

@@ -83,8 +83,8 @@ function train(; kws...)
         WithGraph(fg, GATConv(args.hidden_dim*args.heads=>args.target_dim, heads=args.heads, concat=false)),
     ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/gcn_with_fixed_graph.jl

@@ -78,8 +78,8 @@ function train(; kws...)
         WithGraph(fg, GCNConv(args.hidden_dim=>args.target_dim)),
     ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/gde.jl

@@ -85,8 +85,8 @@ function train(; kws...)
         WithGraph(fg, GCNConv(args.hidden_dim=>args.target_dim)),
     ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model, node.p)

examples/graphconv.jl

@@ -73,8 +73,8 @@ function train(; kws...)
         WithGraph(fg, GraphConv(args.hidden_dim=>args.target_dim)),
     ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/msg-passing.jl

@@ -75,8 +75,8 @@ function train(; kws...)
         WithGraph(fg, GraphConv(fg, args.hidden_dim=>args.target_dim)),
     ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/semisupervised_gcn.jl

@@ -84,8 +84,8 @@ function train(; kws...)
         node_feature,
     ) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)

examples/vgae.jl

@@ -94,8 +94,8 @@ function train(; kws...)
 
     model = VGAE(encoder, decoder) |> device
 
-    # ADAM optimizer
-    opt = ADAM(args.η)
+    # Adam optimizer
+    opt = Adam(args.η)
 
     # parameters
     ps = Flux.params(model)