From f60aae98158f5a1845975fa922cb164726fb5057 Mon Sep 17 00:00:00 2001 From: Kaushik B <45285388+kaushikb11@users.noreply.github.com> Date: Tue, 29 Jun 2021 17:53:48 +0530 Subject: [PATCH] Update `dataloaders` params in example (#8191) --- pl_examples/bug_report_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pl_examples/bug_report_model.py b/pl_examples/bug_report_model.py index abb65ba86f..f906ab9bde 100644 --- a/pl_examples/bug_report_model.py +++ b/pl_examples/bug_report_model.py @@ -59,8 +59,8 @@ def run(): max_epochs=1, weights_summary=None, ) - trainer.fit(model, train_dataloader=train_data, val_dataloaders=val_data) - trainer.test(model, test_dataloaders=test_data) + trainer.fit(model, train_dataloaders=train_data, val_dataloaders=val_data) + trainer.test(model, dataloaders=test_data) if __name__ == '__main__':