質問編集履歴
1
エラーを省略してしまっていた。
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -4,10 +4,14 @@
|
|
|
4
4
|
PyTorchでファインチューニングを行いたいと思っています。
|
|
5
5
|
|
|
6
6
|
### 発生している問題・エラーメッセージ
|
|
7
|
+
Traceback (most recent call last):
|
|
8
|
+
File "Arashi/arashi.py", line 52, in <module>
|
|
9
|
+
datasets = torch.utils.data.TensorDataset(img_datas, labels)
|
|
10
|
+
File "/home/selen/.pyenv/versions/3.7.3/lib/python3.7/site-packages/torch/utils/data/dataset.py", line 36, in __init__
|
|
7
11
|
assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors)
|
|
12
|
+
File "/home/selen/.pyenv/versions/3.7.3/lib/python3.7/site-packages/torch/utils/data/dataset.py", line 36, in <genexpr>
|
|
13
|
+
assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors)
|
|
8
14
|
AttributeError: 'list' object has no attribute 'size'
|
|
9
|
-
```
|
|
10
|
-
|
|
11
15
|
### 該当のソースコード
|
|
12
16
|
import torch, torchvision
|
|
13
17
|
import torch.nn as nn
|