jamescalam commited on
Commit
cfebb04
1 Parent(s): b1741c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.2)
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