Skip pretrained models for <1.7

LuxDL · Jun 8, 2022 · cb58cfa · cb58cfa
1 parent fc38a40
commit cb58cfa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/Boltz/test/vision.jl b/lib/Boltz/test/vision.jl
@@ -65,6 +65,7 @@ models_available = Dict(alexnet => [(:alexnet, true), (:alexnet, false)],
     @time begin @testset "name = $name & pretrained = $pretrained" for (name, pretrained) in config
         if VERSION <= v"1.7" && pretrained
             @warn "Skipping pretrained models in Julia < 1.7"
+            continue
         end
         model, ps, st = model_creator(name; pretrained)
         st = Lux.testmode(st)