Update README.md

README.md

@@ -255,7 +255,7 @@ The objective of RealFormer is to attain the maximum level of detail to the real
 ```python
 RealFormerAGA(
   (patch_embed): DynamicPatchEmbedding(
-    (proj): Conv2d(2048, 768, kernel_size=(1, 1), stride=(1, 1))
+    (proj): Conv2d(3, 768, kernel_size=(1, 1), stride=(1, 1))
   )
   (encoder_layers): ModuleList(
     (0-15): 16 x TransformerEncoderBlock(
@@ -330,9 +330,9 @@ RealFormerAGA(
     (bn): BatchNorm2d(3, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
     (relu): ReLU(inplace=True)
   )
-  (final_layer): Conv2d(3, 2048, kernel_size=(1, 1), stride=(1, 1))
+  (final_layer): Conv2d(3, 3, kernel_size=(1, 1), stride=(1, 1))
   (style_encoder): Sequential(
-    (0): Conv2d(2048, 768, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
+    (0): Conv2d(3, 768, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
     (1): ReLU()
     (2): AdaptiveAvgPool2d(output_size=1)
     (3): Flatten(start_dim=1, end_dim=-1)