kenshohara/3D-ResNets-PyTorch

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