Samuel Schmidt commited on
Commit
b836cf0
·
1 Parent(s): 7db538c

Add: Logging statement

Browse files
Files changed (1) hide show
  1. src/CLIP.py +1 -0
src/CLIP.py CHANGED
@@ -18,6 +18,7 @@ class CLIPImageEncoder:
18
  images = batch["image"]
19
  print(images)
20
  input = self.processor(images=images, return_tensors="pt")
 
21
  with torch.no_grad():
22
  image_features = self.model.get_image_features(**input)
23
  #image_features = self.model(**input).last_hidden_state[:,0].cpu()
 
18
  images = batch["image"]
19
  print(images)
20
  input = self.processor(images=images, return_tensors="pt")
21
+ print(input)
22
  with torch.no_grad():
23
  image_features = self.model.get_image_features(**input)
24
  #image_features = self.model(**input).last_hidden_state[:,0].cpu()