diff --git a/src/Flux.jl b/src/Flux.jl index 4aa4d8e9..c5693f29 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -34,6 +34,4 @@ include("layers/normalisation.jl") include("data/Data.jl") -include("batches/Batches.jl") - end # module diff --git a/src/batches/Batches.jl b/src/batches/Batches.jl deleted file mode 100644 index 066f4d1c..00000000 --- a/src/batches/Batches.jl +++ /dev/null @@ -1,7 +0,0 @@ -module Batches - -import ..Flux - -include("batch.jl") - -end diff --git a/src/batches/batch.jl b/src/batches/batch.jl deleted file mode 100644 index 5a2eb82e..00000000 --- a/src/batches/batch.jl +++ /dev/null @@ -1,8 +0,0 @@ -struct Batch{T,A,M} - data::A - mask::M -end - -Batch{T}(data, mask) where T = Batch{T,typeof(data),typeof(mask)}(data, mask) - -Batch(xs) = Batch{typeof(first(xs))}(Flux.batch(xs),trues(length(xs)))