diff --git a/net/resnet_embedding.py b/net/resnet_embedding.py
index 4516fca..1250395 100644
--- a/net/resnet_embedding.py
+++ b/net/resnet_embedding.py
@@ -56,6 +56,7 @@ class ResNet50EmbeddingNet(nn.Module):
         )
         
         # 图片预处理变换（用于推理）
+        # TODO 这里应该要修改归一化的逻辑，不然不一样。
         self.preprocess = transforms.Compose([
             transforms.Resize((224, 224), interpolation=transforms.InterpolationMode.BILINEAR),
             transforms.ToTensor(),