ludwig icon indicating copy to clipboard operation
ludwig copied to clipboard

test: tv encoders with large memory footprint

Open jimthompson5802 opened this issue 2 years ago • 2 comments

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

jimthompson5802 avatar Feb 11 '23 23:02 jimthompson5802