pytorch – segmentation を TPU で実行してみた / pytorch – lightning で書き換えてみた