Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Lornatang committed Aug 12, 2022
1 parent 0cbbc36 commit 4985676
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model.py
Original file line number Diff line number Diff line change
@@ -17,12 +17,12 @@
from torch import Tensor
from torch import nn


__all__ = [
"AlexNet",
"alexnet",
]


class AlexNet(nn.Module):
def __init__(self, num_classes: int = 1000) -> None:
super(AlexNet, self).__init__()

0 comments on commit 4985676

Please sign in to comment.