diff --git a/test/test_datasets.py b/test/test_datasets.py index a108479aee3..10f3f4794d5 100644 --- a/test/test_datasets.py +++ b/test/test_datasets.py @@ -616,7 +616,6 @@ class VOCSegmentationTestCase(datasets_utils.ImageDatasetTestCase): year=[f"20{year:02d}" for year in range(7, 13)], image_set=("train", "val", "trainval") ), dict(year="2007", image_set="test"), - dict(year="2007-test", image_set="test"), ) def inject_fake_data(self, tmpdir, config): diff --git a/test/test_models.py b/test/test_models.py index 0acef4dcef6..69a89d900e2 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -244,7 +244,13 @@ def _check_input_backprop(model, inputs): # The following contains configuration parameters for all models which are used by # the _test_*_model methods. _model_params = { - "inception_v3": {"input_shape": (1, 3, 299, 299)}, + "inception_v3": { + "input_shape": (1, 3, 299, 299), + "init_weights": True, + }, + "googlenet": { + "init_weights": True, + }, "retinanet_resnet50_fpn": { "num_classes": 20, "score_thresh": 0.01,