Jyothirmai commited on
Commit
299f4de
·
verified ·
1 Parent(s): 303a358

Update tester.py

Browse files
Files changed (1) hide show
  1. tester.py +1 -3
tester.py CHANGED
@@ -81,9 +81,7 @@ class CaptionSampler(object):
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
- transform = transforms.Compose([
85
- transforms.PILToTensor()
86
- ])
87
 
88
  imageData = transform(image_file)
89
  print(imageData)
 
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
+ transform = transforms.Compose([transforms.ToTensor()])
 
 
85
 
86
  imageData = transform(image_file)
87
  print(imageData)