TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mxnet error message test

Browse Source
This commit is contained in:
Mike J Innes 2017-02-21 16:52:24 +00:00
parent 8e726167dc
commit b2b2c209c6
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test/backend.jl
View File

@ -19,6 +19,16 @@ end
@test mx.infer_shape(f.arch, data = (20, 20, 5, 1))[2] == [(10, 1)] @test mx.infer_shape(f.arch, data = (20, 20, 5, 1))[2] == [(10, 1)]
end end
@mxonly let
model = TLP(Affine(10, 20), Affine(21, 15))
info("The following warning is normal")
e = try mxnet(model, (10, 1))
catch e e end
@test e.trace[1].func == Symbol("Flux.Affine")
@test e.trace[2].func == :TLP
end
# TensorFlow # TensorFlow
@tfonly let dt = tf(d) @tfonly let dt = tf(d)