When I run the main.py. I meet this problem. How do I solve this??

[CGL-learner]

Traceback (most recent call last):
File "/root/Micro-Classify/ml_model/src/main.py", line 30, in
train_model(model, train_loader, train_loader, criterion, optimizer, scheduler, device, num_epochs=num_epochs)
File "/root/Micro-Classify/ml_model/src/utils/train.py", line 11, in train_model
for images, labels in train_loader:
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 634, in next
data = self._next_data()
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 678, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataset.py", line 298, in getitem
return self.dataset[self.indices[idx]]
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataset.py", line 298, in getitem
return self.dataset[self.indices[idx]]
File "/root/Micro-Classify/ml_model/src/data/dataset.py", line 116, in getitem
image = aug(image)
File "/root/Micro-Classify/ml_model/src/data/dataset.py", line 74, in call
return method(data)
File "/root/Micro-Classify/ml_model/src/data/dataset.py", line 45, in window_warping
warped_window = torch.nn.functional.interpolate(
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/functional.py", line 3974, in interpolate
raise NotImplementedError("Got 4D input, but linear mode needs 3D input")
NotImplementedError: Got 4D input, but linear mode needs 3D input