モデルの保存 params = net.state_dict() #netはモデル名 torch.save(params, "ファイル名.prm", pickle_protocol=4) モデルの読み込み params = torch.load("ファイル名.prm", map_location="cpu") net.load_state_dict(params)

PytorchでCNN(画像分類)を書いてみる。 以下を参考にCIFAR10 datasetを用いたCNNを書いてみる。 pytorch.org ソースコード import torch import torchvision import torchvision.transforms as transforms import torch.nn as nn import torch.nn.functional…

Raspberry pi 4をアップグレード（カーネル：2020/02/10）したら起動しなくなった。。。 調べてみると、Raspbian Buster(2020-02-05)で再インストールすると起動するみたい。 できれば、バグが改善されて、再インストールせずにちゃんと起動してくれるといい…

Raspberry-pi 4を購入した。 ずっとRaspberry-pi 2を使っていたので、それに比べたらすごく快適に動作する。 TensorFlowやPyTorchなどでディープラーニングを試せるといいな。

Raspberry Pi4に機械学習ライブラリPyTorchをインストールしてみる。 手順 パッケージのアップデート $ sudo apt-get update $ sudo apt-get upgrade 2Gのswap領域を作成する。 PyTorchのビルドにはメモリを大量に使用するので、そのために2Gのswap領域を用…

RaspberryPiにpyenvをインストールしてpython3の環境を構築する。 pyenvのインストール 必要なパッケージのインストール $ sudo apt-get install -y git openssl libssl-dev libbz2-dev libreadline-dev libsqlite3-dev pyenvをgit cloneする $ git clone ht…