error : expected tensor as element 0 in argument 0
sebyo opened this issue · 0 comments
sebyo commented
I tried to execute the script without the spatial transform (set it to None )
and I got the above error ?
the problem is in the loading function in videodataset.py
`def __loading(self, path, frame_indices):
clip = self.loader(path, frame_indices)
if self.spatial_transform is not None:
self.spatial_transform.randomize_parameters()
clip = [self.spatial_transform(img) for img in clip]
clip = torch.stack(clip, 0).permute(1, 0, 2, 3)
return clip`
any suggestion how can I work if I don't want to use the data transformations provided here