Fix code snippet.

JuliaGPU · May 31, 2024 · 8f700b1 · 8f700b1
1 parent 9e6d2e7
commit 8f700b1
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/post/2024-05-28-cuda_5.4.md b/post/2024-05-28-cuda_5.4.md
@@ -53,12 +53,11 @@ n_obs = 300_000
 n_feature = 1000
 X = rand(n_feature, n_obs)
 y = rand(1, n_obs)
-train_data = DataLoader((X, y) |< gpu; batchsize = 2048, shuffle=false)
+train_data = DataLoader((X, y) |> gpu; batchsize = 2048, shuffle=false)
 
-model = Dense(n_feature, >) |< gpu
-loss(m, _x, _y) = Flux.Losses.mse(m(_x), _>)
+model = Dense(n_feature, 1) |> gpu
+loss(m, _x, _y) = Flux.Losses.mse(m(_x), _y)
 opt_state = Flux.setup(Flux.Adam(), model)
-Flux.train!(loss, model, train_data, opt_state)
 for epoch in 1:100
   Flux.train!(loss, model, train_data, opt_state)
 end