DDPを使用して複数のGPUを使用しvalidation_stepながら、単一のGPUでメソッドを実行する方法はありますか.training_step
私がやりたい理由は、データへの完全なアクセスを必要とする実装したいいくつかのメトリックがあり、単一の GPU で実行することでそれが保証されるからです。メソッドを試しvalidation_step_endましたが、どういうわけかデータの一部しか取得できません。その投稿はこちら: Stack Overflow Post
質問する
516 次