Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
Image Captioning gives following error: TypeError: relu(): argument 'input' (position 1) must be Tensor, not InceptionOutputs
Inception net gives 2 outputs. The second output is auxiliary output. So you might want to try - features = self.inception(images)[0]