From fb507aa41167c481743b5aa1956e5b9697edff32 Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Wed, 19 Jul 2023 08:37:26 +0200 Subject: [PATCH] delete file accidentally committed --- cuda.jl | 9 --------- 1 file changed, 9 deletions(-) delete mode 100644 cuda.jl diff --git a/cuda.jl b/cuda.jl deleted file mode 100644 index cb47466120..0000000000 --- a/cuda.jl +++ /dev/null @@ -1,9 +0,0 @@ -using Flux, CUDA - -BN = BatchNorm(3) |> gpu; -x = randn(2, 2, 3, 4) |> gpu; - -NNlib.batchnorm(BN.γ, BN.β, x, BN.μ, BN.σ², BN.momentum; - alpha=1, beta=0, eps=BN.ϵ, - training=Flux._isactive(BN, x)) -