pytorch-lightning
test: move the pretrained model to the same device as current model before assertion
#19668
Merged

test: move the pretrained model to the same device as current model before assertion #19668

carmocca merged 3 commits into Lightning-AI:master from models
fnhirwa
fnhirwa move the pretrained model to the same device as model in unit test
8c559aba
fnhirwa fnhirwa marked this pull request as ready for review 1 year ago
fnhirwa fnhirwa requested a review from awaelchli awaelchli 1 year ago
fnhirwa fnhirwa requested a review from carmocca carmocca 1 year ago
fnhirwa fnhirwa requested a review from justusschock justusschock 1 year ago
fnhirwa fnhirwa requested a review from Borda Borda 1 year ago
github-actions github-actions added pl
awaelchli
awaelchli approved these changes on 2024-03-18
awaelchli awaelchli added tests
awaelchli awaelchli added community
awaelchli awaelchli added this to the 2.2.x milestone 1 year ago
fnhirwa Merge remote-tracking branch 'upstream/master' into models
c655e133
fnhirwa added cpu as traner accelerator
c1b8265a
carmocca
carmocca approved these changes on 2024-03-18
carmocca carmocca merged f4178d5d into master 1 year ago
mergify mergify added ready to be merged

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone