fix block repeats in readme example for Nest

This commit is contained in:
Phil Wang
2022-01-22 21:32:53 -08:00
parent c1528acd46
commit 126d204ff2

View File

@@ -542,7 +542,7 @@ nest = NesT(
dim = 96,
heads = 3,
num_hierarchies = 3, # number of hierarchies
block_repeats = (8, 4, 1), # the number of transformer blocks at each heirarchy, starting from the bottom
block_repeats = (2, 2, 8), # the number of transformer blocks at each heirarchy, starting from the bottom
num_classes = 1000
)