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

Merged

Commits

move the pretrained model to the same device as model in unit test

fnhirwa committed 1 year ago
Merge remote-tracking branch 'upstream/master' into models

fnhirwa committed 1 year ago
added cpu as traner accelerator

fnhirwa committed 1 year ago

Loading