Update README.md

README.md

@@ -51,7 +51,7 @@ class PEFTGuard_T5(nn.Module):
     def __init__(self, device, target_number=3):
         super(PEFTGuard_T5, self).__init__()
         self.device = device
-        self.input_channel = (target_number + 1) * 32
+        self.input_channel = (target_number) * 2 * 24
         self.conv1 = nn.Conv2d(self.input_channel, 32, 8, 8, 0).to(self.device)
         self.fc1 = nn.Linear(256 * 256 * 32, 512).to(self.device)
         self.fc2 = nn.Linear(512, 128).to(self.device)