回答編集履歴
1
追記
test
CHANGED
|
@@ -1 +1,13 @@
|
|
|
1
1
|
labelsがリストであることが原因ではありませんか？ tensorに変換して渡せば良いかと思われます。
|
|
2
|
+
|
|
3
|
+
|
|
4
|
+
|
|
5
|
+
---
|
|
6
|
+
|
|
7
|
+
【追記】下記で出来ませんか？
|
|
8
|
+
|
|
9
|
+
```Python
|
|
10
|
+
|
|
11
|
+
datasets = torch.utils.data.TensorDataset(img_datas, torch.tensor(labels))
|
|
12
|
+
|
|
13
|
+
```
|