Hi,
I am trying to use sparse matrix. When I typing to load the batch I am getting the following error:
for i, data in enumerate(train_loader, 0):
File “/usr/local/lib/python3.5/site-packages/torch/utils/data/dataloader.py”, line 259, in next
batch = self.collate_fn([self.dataset[i] for i in indices])
File “/usr/local/lib/python3.5/site-packages/torch/utils/data/dataloader.py”, line 135, in default_collate
return [default_collate(samples) for samples in transposed]
File “/usr/local/lib/python3.5/site-packages/torch/utils/data/dataloader.py”, line 135, in
return [default_collate(samples) for samples in transposed]
File “/usr/local/lib/python3.5/site-packages/torch/utils/data/dataloader.py”, line 112, in default_collate
return torch.stack(batch, 0, out=out)
File “/usr/local/lib/python3.5/site-packages/torch/functional.py”, line 62, in stack
inputs = [t.unsqueeze(dim) for t in sequence]
File “/usr/local/lib/python3.5/site-packages/torch/functional.py”, line 62, in
inputs = [t.unsqueeze(dim) for t in sequence]
AttributeError: ‘torch.sparse.FloatTensor’ object has no attribute ‘unsqueeze’
The version I am using is 0.3.1.
Thanks,
Ortal