Update README.md

README.md

@@ -117,10 +117,10 @@ transform = create_transform(input_size=input_resolution,
 inputs = transform(image).unsqueeze(0).cuda()
 # model inference
 out_avg_pool, features = model(inputs)
-print("Size of the averaged pool features:", out_avg_pool.size())  # torch.Size([1, 640])
+print("Size of the averaged pool features:", out_avg_pool.size())  # torch.Size([1, 1568])
 print("Number of stages in extracted features:", len(features)) # 4 stages
-print("Size of extracted features in stage 1:", features[0].size()) # torch.Size([1, 80, 56, 56])
-print("Size of extracted features in stage 4:", features[3].size()) # torch.Size([1, 640, 7, 7])
+print("Size of extracted features in stage 1:", features[0].size()) # torch.Size([1, 196, 128, 128])
+print("Size of extracted features in stage 4:", features[3].size()) # torch.Size([1, 1568, 16, 16])
 ```