diff --git a/test/Project.toml b/test/Project.toml index 2a93eb04..f5c9dbb3 100644 --- a/test/Project.toml +++ b/test/Project.toml @@ -7,6 +7,7 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" TestImages = "5e47fb64-e119-507b-a336-dd2b206d9990" TestItemRunner = "f8b46487-2199-4994-9208-9a1283c18c0a" cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd" +Downloads = "f43a241f-c20a-4ad4-852c-f6b1247861c6" [compat] CUDA = "4, 5" diff --git a/test/model_tests.jl b/test/model_tests.jl index 729701bc..7df7a7a4 100644 --- a/test/model_tests.jl +++ b/test/model_tests.jl @@ -2,6 +2,7 @@ using Metalhead, Images, TestImages using Flux: gradient, gpu using CUDA: CUDA, has_cuda +using Downloads export PRETRAINED_MODELS, TEST_FAST, @@ -72,7 +73,9 @@ const TEST_X = let img_array = convert(Array{Float32}, channelview(TEST_IMG)) end # ImageNet labels -const TEST_LBLS = readlines(download("https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")) +const TEST_LBLS = readlines(Downloads.download( + "https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt" +)) function acctest(model) ypred = gpu(model)(TEST_X) |> collect |> vec