YB
Size: a a a
YB
YB
SP
RD
RD
YB
RD
RD
RD
-не работает
YB
RD
AO
AP
SZ
Traceback (most recent call last):
File "/content/train.py", line 76, in <module>
train(model_type='lstm')
File "/content/train.py", line 42, in train
for i_batch, sampled_batch in tqdm(enumerate(train_data_loader)):
File "/usr/local/lib/python3.6/dist-packages/tqdm/std.py", line 1104, in __iter__
for obj in iterable:
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 363, in __next__
data = self._next_data()
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 403, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 47, in fetch
return self.collate_fn(data)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/collate.py", line 74, in default_collate
return {key: default_collate([d[key] for d in batch]) for key in elem}
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/collate.py", line 74, in <dictcomp>
return {key: default_collate([d[key] for d in batch]) for key in elem}
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/collate.py", line 55, in default_collate
return torch.stack(batch, 0, out=out)
RuntimeError: stack expects each tensor to be equal size, but got [48, 1] at entry 0 and [36, 1] at entry 1
SZ
YB
SZ
AK
AL
МП