diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
index d080511978c..9b9711781c4 100644
--- a/torchgeo/trainers/segmentation.py
+++ b/torchgeo/trainers/segmentation.py
@@ -4,7 +4,6 @@
 """Trainers for semantic segmentation."""
 
 import os
-import warnings
 from typing import Any, Optional, Union
 
 import matplotlib.pyplot as plt