raise AssertionError( AssertionError: original output #2 is None, but only the following types are supported: (<class 'torch.Tensor'>, <class 'torch.SymInt'>, <class 'torch.SymFloat'>, <class 'torch.SymBool'>)
can anybody suggest the input size and model class code to be put in the example directory?
current example input is torch.randn(input_shape) i.e.
(self.batch_size,1, self.input_frames)