Skip to content
Discussion options

You must be logged in to vote

Hi
PyTorch Lightning 0.4.6 is extremely old. You should consider upgrading.
That being said, you can always test your model as you would in plain pytorch, because the LightningModule is also just a nn.Module:

for inp, target in test_dataloader:
    pred = model(inp)
    test_loss = loss(pred, target)
    ...

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@tchaton
Comment options

Answer selected by AIhang427
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment