inputs.shape = torch.Size([1, 3, 224, 224])
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-11-f63fb1f7f96f> in <module>()
      6 #   print(inputs.shape)
      7   output = tsn.features(inputs)
----> 8   verb_logits, noun_logits = tsn.logits(output)
      9 
     10 

/root/.cache/torch/hub/epic-kitchens_action-models_master/tsn.py in logits(self, features)
    412             if self.reshape:
    413                 logits_verb = logits_verb.view(
--> 414                     (-1, self.num_segments) + logits_verb.size()[1:]
    415                 )
    416             output_verb = self.consensus(logits_verb)

RuntimeError: shape '[-1, 8, 125]' is invalid for input of size 125