updates + format

This commit is contained in:
Awni Hannun
2023-12-14 12:09:10 -08:00
parent 29b7a97342
commit b1b9b11801
4 changed files with 55 additions and 56 deletions

View File

@@ -59,7 +59,6 @@ class Block(nn.Module):
self.shortcut = None
def __call__(self, x):
out = nn.relu(self.bn1(self.conv1(x)))
out = self.bn2(self.conv2(out))
if self.shortcut is None: