From 3ee326ea85faeffa8f73e84ecf0e6b2fa9a5bd68 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Fri, 5 Mar 2021 18:56:54 -0800 Subject: [PATCH] readme --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 19d63ac..165ebb0 100644 --- a/README.md +++ b/README.md @@ -45,7 +45,7 @@ model = Glom( ) img = torch.randn(1, 3, 224, 224) -all_levels = model(img, iters = 12, return_all = True) # (13, 1, 256, 6, 512) - (batch, patches, levels, dimension) +all_levels = model(img, iters = 12, return_all = True) # (13, 1, 256, 6, 512) - (time, batch, patches, levels, dimension) # get the top level outputs after iteration 6 top_level_output = all_levels[7, :, :, -1] # (1, 256, 512) - (batch, patches, dimension)