From 126d204ff255b5e66880600b0151a79d2a293943 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Sat, 22 Jan 2022 21:32:53 -0800 Subject: [PATCH] fix block repeats in readme example for Nest --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 6503744..1eb3abb 100644 --- a/README.md +++ b/README.md @@ -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 )