Update README.md

README.md

@@ -44,7 +44,7 @@ time_reduced_hidden_states = all_layer_hidden_states.mean(-2)
 print(time_reduced_hidden_states.shape) # [13, 768]
 
 # you can even use a learnable weighted average representation
-aggregator = nn.Conv1d(in_channels=12, out_channels=1, kernel_size=1)
+aggregator = nn.Conv1d(in_channels=13, out_channels=1, kernel_size=1)
 weighted_avg_hidden_states = aggregator(time_reduced_hidden_states).squeeze()
 ```