remove batching work for now
This commit is contained in:
parent
2b545cf0ca
commit
5118ef9163
@ -34,6 +34,4 @@ include("layers/normalisation.jl")
|
|||||||
|
|
||||||
include("data/Data.jl")
|
include("data/Data.jl")
|
||||||
|
|
||||||
include("batches/Batches.jl")
|
|
||||||
|
|
||||||
end # module
|
end # module
|
||||||
|
@ -1,7 +0,0 @@
|
|||||||
module Batches
|
|
||||||
|
|
||||||
import ..Flux
|
|
||||||
|
|
||||||
include("batch.jl")
|
|
||||||
|
|
||||||
end
|
|
@ -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)))
|
|
Loading…
Reference in New Issue
Block a user