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