Skip to content

Commit

Permalink
Compat bounds for julia, Knet, SpecialFunctions
Browse files Browse the repository at this point in the history
  • Loading branch information
IanButterworth committed Sep 19, 2019
1 parent 474d0bb commit bab901f
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 2 deletions.
6 changes: 6 additions & 0 deletions Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,12 @@ ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
QuartzImageIO = "dca85d43-d64c-5e67-8c65-017450d5d020"
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
Requires = "ae029012-a4dd-5104-9daa-d747884805df"
SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"

[compat]
julia = "0.7, 1.0"
SpecialFunctions = "≥ 0.8.0"
Knet = "≥ 1.2.5"

[extras]
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
Expand Down
6 changes: 4 additions & 2 deletions test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -83,9 +83,11 @@ end
YOLO.loadWeights!(model, settings)

res = model(vocloaded.imstack_mat) #run once to deal with compillation overhead
t = @elapsed model(vocloaded.imstack_mat)
t = @elapsed for i in 1:10
model(vocloaded.imstack_mat)
end

inference_time = t / num_images
inference_time = (t / 10) / num_images
inference_rate = 1 / inference_time
@test inference_time < 1.0 #seconds

Expand Down

0 comments on commit bab901f

Please sign in to comment.