ludwig
ludwig copied to clipboard
test: tv encoders with large memory footprint
Code Pull Requests
Add unit test for TorchVision pretrained models that have a large memory footprint. These large memory models are excluded from the existing unit tests. This new test accounts for the excluded models and allows running the test on a server with a suitable configuration.
This test has the custom pytest marker large_memory