diff --git a/encoder.py b/encoder.py index cdfa384cbf34823c764a4484cff228f34452b34f..c888b22afea7fb300c3a8ff998c83ba802b926e6 100644 --- a/encoder.py +++ b/encoder.py @@ -41,7 +41,8 @@ import timm # from torchgeo.datasets import RasterDataset, BoundingBox,stack_samples # from torchgeo.samplers import GridGeoSampler, Units # from torchgeo.transforms import AugmentationSequential -# from .utils.torchgeo import NoBordersGridGeoSampler +## from .utils.torchgeo import NoBordersGridGeoSampler +# from .utils.trchg import NoBordersGridGeoSampler from .utils.geo import get_mean_sd_by_band from .utils.geo import merge_tiles @@ -55,7 +56,7 @@ from .utils.misc import (QGISLogHandler, compute_md5_hash, log_parameters_to_csv, ) -from .utils.torch import quantize_model +from .utils.trch import quantize_model from .tg.datasets import RasterDataset from .tg.utils import stack_samples, BoundingBox diff --git a/utils/torch.py b/utils/trch.py similarity index 100% rename from utils/torch.py rename to utils/trch.py diff --git a/utils/torchgeo.py b/utils/trchg.py similarity index 100% rename from utils/torchgeo.py rename to utils/trchg.py