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("batches/Batches.jl")
|
||||
|
||||
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