TST Speed up vision model tests (#2058)
The HRA vision model test is extremely slow on CI (> 600 sec, 50% of
total time). This change speeds up the test by using a smaller ResNet
model to run the tests.
It's still not clear why HRA was so slow specifically -- LoRA is 40x
faster -- but that can be fixed separately.