PyTorch
記事
- PyTorch Hub, torchvision.modelsで学習済みモデルをダウンロード・使用
- PyTorchの学習済みモデルで画像分類(VGG, ResNetなど)
- PyTorchでTensorとモデルのGPU / CPUを指定・切り替え
- PyTorchでGPU情報を確認(使用可能か、デバイス数など)
- PyTorchのTensorのデータ型(dtype)と型変換(キャスト)
- PyTorchでモデル(ネットワーク)を構築・生成
- PyTorchのtorch.flattenとtorch.nn.Flattenの違い
- PyTorchのTensorの次元数、形状、要素数を取得: dim(), size(), numel()
- PyTorchのTensorの要素の値を取得: item()