fixed submit attempt #1 (failed to parse tuple type hint)
This commit is contained in:
parent
39aa685e28
commit
c4a93e9fae
|
|
@ -4,8 +4,7 @@ import torch.nn
|
|||
|
||||
class MyCNN(torch.nn.Module):
|
||||
def __init__(self,
|
||||
input_channels: int,
|
||||
input_size: tuple[int, int]):
|
||||
input_channels: int):
|
||||
super().__init__()
|
||||
|
||||
self.layers = torch.nn.Sequential(
|
||||
|
|
@ -51,4 +50,4 @@ class MyCNN(torch.nn.Module):
|
|||
return str(self.layers)
|
||||
|
||||
|
||||
model = MyCNN(input_channels=1, input_size=(100, 100))
|
||||
model = MyCNN(input_channels=1)
|
||||
|
|
|
|||
|
|
@ -4,8 +4,7 @@ import torch.nn
|
|||
|
||||
class MyCNN(torch.nn.Module):
|
||||
def __init__(self,
|
||||
input_channels: int,
|
||||
input_size: tuple[int, int]):
|
||||
input_channels: int):
|
||||
super().__init__()
|
||||
|
||||
self.layers = torch.nn.Sequential(
|
||||
|
|
@ -51,4 +50,4 @@ class MyCNN(torch.nn.Module):
|
|||
return str(self.layers)
|
||||
|
||||
|
||||
model = MyCNN(input_channels=1, input_size=(100, 100))
|
||||
model = MyCNN(input_channels=1)
|
||||
|
|
|
|||
Loading…
Reference in New Issue