Загрузка данных в torch.utils.data.TensorDataset - PullRequest
Новая
       31

Загрузка данных в torch.utils.data.TensorDataset

0 голосов
/ 04 декабря 2018

Я хотел бы запустить модель ОЗУ на своем собственном наборе данных.Мой тренировочный набор - train.t7.Я столкнулся с проблемой при загрузке набора данных, потому что я впервые использую Python.Я очень старался понять и программное обеспечение, и пакет, но мне нужна ваша помощь в решении этой проблемы.

Я запускаю следующий код: 

o=torchfile.load('train.t7')
x=tf.convert_to_tensor(o[b'data'],dtype=tf.uint8)
y=tf.convert_to_tensor(o[b'labels'],dtype=tf.uint8)
x=tf.reshape(x,[50000,784])
dataset = torch.utils.data.TensorDataset(x, y)

Но я получил следующую ошибку:

File "/home/wing92518/.local/lib/python3.6/site-packages/torch/utils/data/dataset.py", line 36, in <genexpr>
    assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors)
AttributeError: 'Tensor' object has no attribute 'size'
...