Skip to content

Commit e015933

Browse files
committed
🔥 Remove large models from tests on GitHub actions.
1 parent ec2f431 commit e015933

File tree

1 file changed

+0
-14
lines changed

1 file changed

+0
-14
lines changed

tests/engines/test_patch_predictor.py

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -305,21 +305,7 @@ def test_patch_predictor_kather100k_output(
305305
pretrained_info = {
306306
"alexnet-kather100k": [1.0, 0.9999735355377197],
307307
"resnet18-kather100k": [1.0, 0.9999911785125732],
308-
"resnet34-kather100k": [1.0, 0.9979840517044067],
309-
"resnet50-kather100k": [1.0, 0.9999986886978149],
310-
"resnet101-kather100k": [1.0, 0.9999932050704956],
311-
"resnext50_32x4d-kather100k": [1.0, 0.9910059571266174],
312-
"resnext101_32x8d-kather100k": [1.0, 0.9999971389770508],
313-
"wide_resnet50_2-kather100k": [1.0, 0.9953408241271973],
314-
"wide_resnet101_2-kather100k": [1.0, 0.9999831914901733],
315-
"densenet121-kather100k": [1.0, 1.0],
316-
"densenet161-kather100k": [1.0, 0.9999959468841553],
317-
"densenet169-kather100k": [1.0, 0.9999934434890747],
318-
"densenet201-kather100k": [1.0, 0.9999983310699463],
319-
"mobilenet_v2-kather100k": [0.9999998807907104, 0.9999126195907593],
320-
"mobilenet_v3_large-kather100k": [0.9999996423721313, 0.9999878406524658],
321308
"mobilenet_v3_small-kather100k": [0.9999998807907104, 0.9999997615814209],
322-
"googlenet-kather100k": [1.0, 0.9999639987945557],
323309
}
324310
for model, expected_prob in pretrained_info.items():
325311
_test_predictor_output(

0 commit comments

Comments
 (0)