readme

lucidrains · Mar 6, 2021 · 3ee326e · 3ee326e
1 parent b3b6b87
commit 3ee326e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)