c96649bb6e1dfeb9479de6c5e8547d510c6b7a32,gluoncv/auto/tasks/utils.py,,auto_suggest,#Any#Any#Any#,80

Before Change

    elif isinstance(estimator, (tuple, list)):
        pass
    else:
        assert issubclass(estimator, BaseEstimator)
        estimator = [estimator]
    config["estimator"] = ag.Categorical(*estimator)

def get_recursively(search_dict, field):

After Change

        pass
    else:
        if isinstance(estimator, ag.Space):
            estimator = estimator.data
        elif isinstance(estimator, str):
            estimator = [estimator]
        for i, e in enumerate(estimator):
            if e == "ssd":
                estimator[i] = SSDEstimator
            elif e == "yolo3":
                estimator[i] = YOLOv3Estimator
            elif e == "faster_rcnn":
                estimator[i] = FasterRCNNEstimator
            elif e == "center_net":
                estimator[i] = CenterNetEstimator
        config["estimator"] = ag.Categorical(*estimator)

    // get dataset statistics
    // user needs to define a Dataset object "train_dataset" when using custom dataset

Instances