Q&A
現在,こちらのpytorch-lightningのtutorialで勉強を行なっております.
TUTORIAL 5: TRANSFORMERS AND MULTI-HEAD ATTENTION
こちらに関して質問があります.
セル33に関してtutorialではtestデータに関してはサンプルを表示しているのですが,trainデータに対してはサンプルを表示していないため,表示したいと考え,下記のコードを実装してみました:
python
1pl.seed_everything(42) 2x, indices, y = next(iter(train_anom_loader)) 3visualize_exmp(indices[:4], train_set)
しかし,表示されるのは下記のようなものです.
明らかにlabelの種類が異なっており,以下のコードでいくつかの画像に関してlabelを確認してみても一致していませんでした.
python
1c=0 2for i in indices.reshape(-1): 3 print(i,train_set[i][1]) 4 c+=1 5 if c>12: 6 break
こちらどのようにしたらtrainのサンプルを表示させることができるかご教授して頂けると幸いです。
補足 :
今回の内容に関する実装はgoogle colaboratory上で
python
1!pip install pytorch_lightning
をした後にセル2,3および
セル26-33までを実装することで再現できます.
あなたの回答
tips
プレビュー
