these are now useless
This commit is contained in:
parent
536949891d
commit
8869f1f953
@ -36,7 +36,6 @@ include("layers/affine.jl")
|
||||
include("layers/activation.jl")
|
||||
include("layers/cost.jl")
|
||||
include("layers/recurrent.jl")
|
||||
include("layers/shims.jl")
|
||||
|
||||
include("data.jl")
|
||||
include("training.jl")
|
||||
|
@ -1,37 +0,0 @@
|
||||
struct Conv2D
|
||||
filter::Param{Array{Float64,4}} # [height, width, inchans, outchans]
|
||||
stride::Dims{2}
|
||||
end
|
||||
|
||||
Conv2D(size; in = 1, out = 1, stride = (1,1), init = initn) =
|
||||
Conv2D(param(initn(size..., in, out)), stride)
|
||||
|
||||
infer(c::Conv2D, in::Dims{4}) =
|
||||
(in[1], map(i -> (in[i+1]-size(c.filter,i))÷c.stride[i]+1, (1,2))..., size(c.filter, 4))
|
||||
|
||||
# TODO: many of these should just be functions
|
||||
|
||||
for Pool in :[MaxPool, AvgPool].args
|
||||
@eval begin
|
||||
struct $Pool
|
||||
size::Dims{2}
|
||||
stride::Dims{2}
|
||||
end
|
||||
|
||||
$Pool(size; stride = (1,1)) =
|
||||
$Pool(size, stride)
|
||||
|
||||
infer(c::$Pool, in::Dims{4}) =
|
||||
(in[1], map(i -> (in[i+1]-c.size[i])÷c.stride[i]+1, (1,2))..., in[4])
|
||||
|
||||
shape(c::$Pool) = nothing
|
||||
end
|
||||
end
|
||||
|
||||
struct Reshape{N}
|
||||
dims::Dims{N}
|
||||
end
|
||||
|
||||
Reshape(dims::Integer...) = Reshape(dims)
|
||||
|
||||
shape(r::Reshape, ::Void) = r.dims
|
Loading…
Reference in New Issue
Block a user