Я хотел бы запустить модель ОЗУ на своем собственном наборе данных.Мой тренировочный набор - 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'