s487179/x1
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

train

Browse Source
This commit is contained in:
wojciechbatruszewicz 2023-06-26 20:16:56 +02:00
parent a979df9218
commit 8e066818ea
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -11,7 +11,7 @@ import argparse
class MyNeuralNetwork(nn.Module): class MyNeuralNetwork(nn.Module):
def __init__(self, *args, **kwargs) -> None: def __init__(self, *args, **kwargs) -> None:
super(MyNeuralNetwork, self).__init__(*args, **kwargs) super(MyNeuralNetwork, self).__init__(*args, **kwargs)
self.fc1 = nn.Linear(12, 64) self.fc1 = nn.Linear(32, 7)
self.relu = nn.ReLU() self.relu = nn.ReLU()
self.fc1 = nn.Linear(12, 64) self.fc1 = nn.Linear(12, 64)
self.relu = nn.ReLU() self.relu = nn.ReLU()