Skip to content

Commit 952e34e

Browse files
fix(tests): reduce r2plus1d_18 input size to prevent OOM
The test_video_r2plus1_18 test used a (16,3,16,112,112) input (~96M elements) which consistently got OOM-killed. Reduced to (1,3,1,112,112) which passes reliably while still exercising the full model. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent af0d19a commit 952e34e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_real_world_models_slow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -832,7 +832,7 @@ def test_taskonomy(default_input1):
832832

833833
def test_video_r2plus1_18():
834834
model = torchvision.models.video.r2plus1d_18()
835-
model_input = torch.randn(16, 3, 16, 112, 112)
835+
model_input = torch.randn(1, 3, 1, 112, 112)
836836
show_model_graph(
837837
model,
838838
model_input,

0 commit comments

Comments
 (0)