Skip to content

Commit

Permalink
Remove obsolete parameter (#1572)
Browse files Browse the repository at this point in the history
  • Loading branch information
charlesbvll authored Jan 14, 2023
1 parent 8d4b107 commit cc40177
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions examples/quickstart_pytorch_lightning/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,15 +24,15 @@ def set_parameters(self, parameters):
def fit(self, parameters, config):
self.set_parameters(parameters)

trainer = pl.Trainer(max_epochs=1, progress_bar_refresh_rate=0)
trainer = pl.Trainer(max_epochs=1)
trainer.fit(self.model, self.train_loader, self.val_loader)

return self.get_parameters(config={}), 55000, {}

def evaluate(self, parameters, config):
self.set_parameters(parameters)

trainer = pl.Trainer(progress_bar_refresh_rate=0)
trainer = pl.Trainer()
results = trainer.test(self.model, self.test_loader)
loss = results[0]["test_loss"]

Expand Down
2 changes: 1 addition & 1 deletion examples/quickstart_pytorch_lightning/mnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ def main() -> None:
model = LitAutoEncoder()

# Train
trainer = pl.Trainer(max_epochs=5, progress_bar_refresh_rate=0)
trainer = pl.Trainer(max_epochs=5)
trainer.fit(model, train_loader, val_loader)

# Test
Expand Down

0 comments on commit cc40177

Please sign in to comment.