グーグルコラボで今まで動いていたプログラムが、動かなくなりました

Question

前提

ここに質問の内容を詳しく書いてください。
画像を2群に判別するプログラムを書いています

実現したいこと

最近まで動いていたのに
給に動かなくなりました

発生している問題・エラーメッセージ

---

AttributeError Traceback (most recent call last)
<ipython-input-9-689a7ef2904d> in <module>
54
55 prediction_loader = torch.utils.data.DataLoader(prediction_subset, batch_size=batch_size, shuffle=True)
---> 56 inputs, _ = iter(prediction_loader).next()
57
58 #inputs, _ = iter(prediction_loader).next(iter(dataloader))

AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'

該当のソースコード

data_type = "学習データ"
batch_size = 8

モデルの選択 = "自作のモデル"

入力画像を表示する関数の作成

def imshow(img):
img = img / 2 + 0.5
plt.title("予測に使った画像")
plt.imshow(img.transpose(0, 1).transpose(1, 2))
plt.axis("off")
plt.show()

def prediction(model, inputs):

imshow(torchvision.utils.make_grid(inputs))
outputs = model(inputs.to(device))
outputs = outputs.to("cpu").argmax(dim=1)

数値から名称に変換

ans = []
sorted_keys = sorted(keywords)
for label in outputs:
for index in range(len(keywords)):
if label==index:
ans.append(sorted_keys[index])
print("AIの予測結果", ans)

if data_type=="学習データ":
prediction_subset = train_subset
else :
prediction_subset = test_subset
model = cnn if モデルの選択 =="自作のモデル" else transfer_model

データローダー

prediction_loader = torch.utils.data.DataLoader(prediction_subset, batch_size=predict_num, shuffle=True)

データローダーから1バッチ分データを取り出す

inputs, _ = iter(prediction_loader).next()

予測する

prediction(cnn, inputs)

補足情報（FW/ツールのバージョンなど）

google コラボラトリを用いています
inputs, _ = iter(prediction_loader).next()でエラーが起きるのですが
昨年まではこんなところでエラーが起きず、バージョン変更によるエラーなのでしょうk、、、

Comments

[meg_]

Pytorchをお使いでしょうか？ GoogleColaboratoryのデフォルトのまま使っているならPytorchのバージョンが変わっていたりしませんか？一度確認されてはどうでしょうか？

[退会済みユーザー]

[最初の投稿：2023/01/29 17:18 の記載内容に対するコメント]
質問に書いてある「該当のソースコード」と「エラーメッセージ」の内容に齟齬があります。

そのエラーメッセージによると、エラーを起こしているのは
inputs, _ = iter(prediction_loader).next()
であって、
inputs, _ = iter(prediction_loader).next(iter(dataloader))
ではありません。

（質問文のコードは
inputs, _ = iter(prediction_loader).next(iter(dataloader))
となっています）

→→→　本当に、質問文のコード（があるセル）を実行されていますか？

一度冷静になって、状況を確認の上、質問内容に修正がある場合は質問文を編集してください

[nact]

お忙しい中、ありがとうございます

meg様
pytorchのバージョンは確認しましたが、前のバージョンが分からず、、、
Python３．８だったので、3.71に変更しましたが、直らず、、、

qnoir様
お忙しい中、ありがとうございます
１行ごとに確認したのですが、ここのinputsで止まってしまいます
ここでエラーが起きているとおもうのですが
よく調べてみたいと思います
ありがとうございます

[退会済みユーザー]

それか、同じセルであっても、質問文に記載してある部分以外のところで起きている可能性があります。
そのセルのコードを省略せずに全部記載してください。

Answer 1

# データローダーから1バッチ分データを取り出す
inputs, _ = iter(prediction_loader).next()

を

inputs, _ = next(iter(prediction_loader))

に変えてみて下さい。

Answer 2

ありがとうございます！
無事に解決しました！