エラー対処法「Found dtype Long but expected Float」 # pytorch_Lighning

pytorch lightnin使用中に、 タイトルのエラーが出ました。

原因

正解ラベルの型が、floatで無かった。

対策

labelをfloatにする。 整数値で良くても、floatに型変換しておく。 下記、イメージ。

float(label)

github.com