diff --git a/torchgeo/trainers/detection.py b/torchgeo/trainers/detection.py index 7cea880d0d3..9602d682de2 100644 --- a/torchgeo/trainers/detection.py +++ b/torchgeo/trainers/detection.py @@ -21,9 +21,6 @@ from ..datasets.utils import unbind_samples -plt.switch_backend("agg") - - BACKBONE_LAT_DIM_MAP = { "resnet18": 512, "resnet34": 512,