Trainer.test() in pytorch ligtning won't work

1

According to the explanation in PYTORCH LIGHTNING DOCUMENTATION, to test the model with a new dataset I should do this:

test = DataLoader(…)
trainer.test(test_dataloaders=test)

but then I get that error:

test() got an unexpected keyword argument ‘test_dataloader’

2

Are you using test_dataloader (expected) or test_dataloaders?

1 Like