Skip to content

Commit 7a32bb0

Browse files
committed
Adjust ConvNeXt and ConvMixer
1 parent b9e238c commit 7a32bb0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/convnets.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ GC.safepoint()
236236
GC.gc()
237237

238238
@testset "ConvNeXt" verbose = true begin
239-
@testset for mode in [:small, :base, :large] # :tiny, #, :xlarge]
239+
@testset for mode in [:small, :base] #, :large # :tiny, #, :xlarge]
240240
@testset for drop_path_rate in [0.0, 0.5]
241241
m = ConvNeXt(mode; drop_path_rate)
242242
@test size(m(x_224)) == (1000, 1)
@@ -251,7 +251,7 @@ GC.safepoint()
251251
GC.gc()
252252

253253
@testset "ConvMixer" verbose = true begin
254-
@testset for mode in [:small, :base, :large]
254+
@testset for mode in [:small, :base] #, :large]
255255
m = ConvMixer(mode)
256256

257257
@test size(m(x_224)) == (1000, 1)

0 commit comments

Comments
 (0)