Enable some ViT tests

FluxML · Apr 4, 2022 · d51e3a4 · d51e3a4
1 parent 6d0d71c
commit d51e3a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -25,6 +25,6 @@ end
 GC.gc()
 
 # ViT tests
-# @testset verbose = true "ViTs" begin
-#   include("vit-based.jl")
-# end
+@testset verbose = true "ViTs" begin
+  include("vit-based.jl")
+end
diff --git a/test/vit-based.jl b/test/vit-based.jl
@@ -2,7 +2,7 @@ using Metalhead, Test
 using Flux
 
 @testset "ViT" begin
-  for mode in [:tiny, :small, :base, :large, :huge, :giant, :gigantic]
+  for mode in [:tiny, :small, :base, :large] #,:huge, :giant, :gigantic]
     m = ViT(mode)
     @test size(m(rand(Float32, 256, 256, 3, 2))) == (1000, 2)
     @test_skip gradtest(m, rand(Float32, 256, 256, 3, 2))