@@ -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