Tsubatoの発信記録

主に機械学習やデータサイエンス関連で学んだことを書いています。

tips

onnxファイルからパラメータ(weight, bias)を抽出する

やりたいこと ニューラルネットワークの表現形式の一つであるonnxファイルからパラメータを取り出したい。 PyTorchなどのフレームワークで一旦読み込んでからパラメータを取り出すこともできるかと思いますが、onnxだけで完結する方法を共有します。 実装 サ…

google colabのTPUをPyTorchで使う

問題 google colabのTPUをPytorchで使いたい。 ググって出てくるセットアップをするとエラーが出る。 # !pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.9-cp37-cp37m-linux_x86_64.whl ERROR: torch_x…

pytestでテスト入力をテスト関数間で共有する

0. この投稿の概要 1. 実装例 テスト対象のクラス パターン1: 1つのテスト関数で複数の関数をテストする。 パターン2: parametrizeのindirectを使う。 パターン3: getfixturevalueを使う。 0. この投稿の概要 クラスメソッドをテストする時に、同じ入力を異…