Spaces:
Runtime error
Runtime error
jamescalam
commited on
Commit
•
cfebb04
1
Parent(s):
b1741c9
Update app.py
Browse files
app.py
CHANGED
@@ -70,7 +70,7 @@ class Classifier:
|
|
70 |
self.model.weight = torch.nn.Parameter(init_weight)
|
71 |
# init loss and optimizer
|
72 |
self.loss = torch.nn.BCEWithLogitsLoss()
|
73 |
-
self.optimizer = torch.optim.SGD(self.model.parameters(), lr=0.
|
74 |
|
75 |
def fit(self, X: list, y: list, iters: int = 5):
|
76 |
# convert X and y to tensor
|
|
|
70 |
self.model.weight = torch.nn.Parameter(init_weight)
|
71 |
# init loss and optimizer
|
72 |
self.loss = torch.nn.BCEWithLogitsLoss()
|
73 |
+
self.optimizer = torch.optim.SGD(self.model.parameters(), lr=0.1)
|
74 |
|
75 |
def fit(self, X: list, y: list, iters: int = 5):
|
76 |
# convert X and y to tensor
|