mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-01 04:14:38 +08:00
updates + format
This commit is contained in:
@@ -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:
|
||||
|
Reference in New Issue
Block a user