juanpablomesa commited on
Commit
83cd9c3
·
1 Parent(s): 41bc78a

Added back some logging messages

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -140,14 +140,14 @@ class EndpointHandler:
140
  frame_embedding = self.model.get_video_features(**frame_preprocessed)
141
 
142
  # Check the shape of the tensor
143
- # self.logger.info(f"Shape of the batch_emb tensor: {frame_embedding.shape}")
144
 
145
  # Normalize the embeddings if it's a 2D tensor
146
  if frame_embedding.dim() == 2:
147
- # self.logger.info("Normalizing embeddings")
148
  batch_emb = torch.nn.functional.normalize(frame_embedding, p=2, dim=1)
149
  else:
150
- # self.logger.info("Skipping normalization due to tensor shape")
151
  batch_emb = frame_embedding.squeeze(0)
152
 
153
  self.logger.info("Converting into numpy array")
 
140
  frame_embedding = self.model.get_video_features(**frame_preprocessed)
141
 
142
  # Check the shape of the tensor
143
+ self.logger.info(f"Shape of the batch_emb tensor: {frame_embedding.shape}")
144
 
145
  # Normalize the embeddings if it's a 2D tensor
146
  if frame_embedding.dim() == 2:
147
+ self.logger.info("Normalizing embeddings")
148
  batch_emb = torch.nn.functional.normalize(frame_embedding, p=2, dim=1)
149
  else:
150
+ self.logger.info("Skipping normalization due to tensor shape")
151
  batch_emb = frame_embedding.squeeze(0)
152
 
153
  self.logger.info("Converting into numpy array")