テクめも

プログラミング関連のちょっとしたTipsなどを書いています。

PyTorch

PyTorchでGlobal Max PoolingとGlobal Average Poolingを使う方法

はじめに Global Max PoolingやGlobal Average Poolingを使いたいとき、KerasではGlobalAveragePooling1Dなどを用いると簡単に使うことができますが、PyTorchではそのままの関数はありません。 そこで、PyTorchでは、Global Max PoolingやGlobal Average Poo…

PyTorchで同期処理を行う

PyTorch Advent Calendar 2018 2日目の記事です。 PyTorchでは*1、GPUの処理は非同期で実施されます。 なので、 t = time.time() model(var) # GPUでの推論 print(time.time() - t) としても、非同期で処理が行われるので正しく処理時間は計算できません。 C…