In [1]:
using Mocha
using PyPlot
using WaveletScattering
ws = WaveletScattering
Out[1]:
In [19]:
spec = ws.Spec1D(log2_size = 13)
bank = ws.Bank1D(spec, max_log2_stride = 0)
data = randn(1<<13)
backend = Mocha.CPUBackend()
signal = ws.InputLayer(
data = data,
tops = [:signal],
symbols = [:time])
layers = Mocha.Layer[signal]
Mocha.init(backend)
@time net = Mocha.Net("network", backend, layers);
In [50]:
exp_dir = "snapshots"
solver_method = SGD()
params = make_solver_parameters(solver_method, max_iter=10000, regu_coef=0.0005,
mom_policy=MomPolicy.Fixed(0.9),
lr_policy=LRPolicy.Inv(0.01, 0.0001, 0.75),
load_from=exp_dir)
solver = Solver(solver_method, params)
Out[50]:
In [51]:
Mocha.solve(solver, net)
In [ ]: