SageMakerでPyTorchのモデルを学習させる - sambaiz-netAWSの機械学習サービスSageMakerでPyTorchのモデルを学習させる。 コード まず学習させるモデルとそれを呼び出すエントリーポイントになるコードを書く。全体のコードはGitHubにある。 実際の環境と同じSageMakerのコンテナをローカルで動かしてVSCodeのRemote Developmentで接続して開発すると入っていないパッケージは警告が出たりして良い。 VSCodeのRemote DevelopmentでSageMakerのコンテナ環境でモデルを開発する - sambaiz-net モデル 以前作ったMNISTのモデルを使う。 PyTorchでMNISTする - sambaiz-net $ cat model.py import torch from torch import nn, cuda import torch.nn.functional as F imp
