損失関数は次のように私は損失関数のRMSEを定義しています

Question

をint型：model.pred

model.pred= tf.argmax(model.props, 1) 
model.actual = tf.argmax(model.y, 1) 
model.RMSE = tf.sqrt(tf.reduce_mean(tf.square(tf.sub(model.predictedSteer, model.actualSteer)))) 

はint64型であるが、それはフロート入力を必要とするため、SQRTは、コンパイルエラーになります。 tf.castを使うと、損失関数は微分できなくなります。これをどうすれば解決できますか？

Answer 1

argmaxは区別できないため、それを含むほとんどの機能も区別できません。この問題についてもっと知ることなく、解決策を提示することは困難です。分類問題の場合は、argmaxではなくロジット（pred）を使用してください。