More than 1 year has passed since last update.

Pytorchモデルの保存、読み込み

Posted at 2022-04-15

保存

torch.save(model.state_dict(), PATH)

読み込み

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()

🐣

フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com

Core MLやARKitを使ったアプリを作っています。
機械学習／AR関連の情報を発信しています。