making x Float32 in migrate from Flux example (#97)

LuxDL · Sep 8, 2022 · 138d824 · 138d824
1 parent 846a82c
commit 138d824
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/src/manual/migrate_from_flux.md b/docs/src/manual/migrate_from_flux.md
@@ -14,7 +14,7 @@ make in the following example.
 
     model = Chain(Dense(2 => 4), BatchNorm(4, relu), Dense(4 => 2))
     rng = Random.default_rng()
-    x = randn(rng, 2, 4)
+    x = randn(rng, Float32, 2, 4)
   
     ps, st = Lux.setup(rng, model)
 
@@ -30,7 +30,7 @@ make in the following example.
 
     model = Chain(Dense(2 => 4), BatchNorm(4, relu), Dense(4 => 2))
     rng = Random.default_rng()
-    x = randn(rng, 2, 4)
+    x = randn(rng, Float32, 2, 4)
 
 
 
@@ -205,4 +205,4 @@ x = randn(rng, 3, 1)
 ps, st = Lux.setup(rng, lmodel)
 
 lmodel(x, ps, st)[1] == fmodel(x)
-```
+```