PyTorch

スポンサーリンク
Machine Learning

【全クリア】強化学習(DQN)でAtari Breakoutを動かす【PyTorch】

前回、GymnasiumでAtari Breakout(ブロック崩し)をランダムに動かすことができたので、ここではPyTorchによる強化学習(DQN)によってゲームを攻略していきます。最終的に、全ブロックを崩すことに成功しました。関連記事...
Machine Learning

【PyTorch】モデルの重みに初期値を設定する方法

PyTorchモデルの重みに初期値を設定する方法について記載していきます。今回、対象とするモデルはシンプル構造の下記コードとします。class Model(nn.Module): def __init__(self): ...
Python

PyTorchで最頻値を算出するtorch.mode

PyTorchで最頻値を算出するにはtorch.modeを使う。torch.mode — PyTorch 1.13 documentationtorch.modeの使い方torch.modeの第一引数inputにTensor配列を指定すると...
Machine Learning

【初心者】Amazon SageMaker Studio Labに登録して、PyTorchをGPUで動かす【機械学習】

誰でも無料でGPUを利用できるAmazon SageMaker Studio Lab(Studio Lab)の始め方の手順を記載していきます。また、GPUでPyTorchを動作させるところまで見ていきます。Studio Labは、AWSのア...
Python

【PyTorch】共分散行列と相関係数行列を算出するtorch.cov, torch.corrcoef

PyTorchで共分散行列と相関係数行列を算出するには、torch.cov、torch.corrcoefを使う。torch.cov — PyTorch 1.12 documentationtorch.corrcoef — PyTorch 1...
Python

【PyTorch】分位数を算出するtorch.quantile

PyTorchで分位数を算出するにはtorch.quantileを使う。箱ひげ図などにも用いられる四分位数もtorch.quantileを使うことで、算出できる。torch.quantile - PyTorch Docs v1.12torc...
Machine Learning

【PyTorch】音声/音へのデータ拡張をするFrequencyMasking、TimeMasking、TimeStretch【SpecAugment】

音声強調、音声認識や音環境分類のような音データにおけるデータ拡張方法であるSpecAugmentをPyTrochで試す。PyTorchには、FrequencyMasking、TimeMaskingとTimeStretchの3つのクラスが用意...
Machine Learning

PyTorchでモデルのパラメータ(重み)を固定する方法、転移学習やファインチューニングに利用可能

転移学習やファインチューニングを行う際に、モデルの重みを固定する必要があり、PyTorchにおける方法を調べてました。モデルのパラメータ(重み)を固定する方法PyTorchの公式Quickstartの例から、簡単なモデルを生成します。関連記...
Python

PyTorchで高速フーリエ変換をするtorch.fft.fft

PyTorchで高速フーリエ変換(離散フーリエ変換)をするには、torch.fft.fftを使う。torch.fft.fftの使い方と、正弦波に対してFFTを行い、周波数特性を確認する。torch.fft.fft - PyTorch Doc...
Python

Tensor配列の非ゼロ要素のインデックスを取得するtorch.nonzero【PyTorch】

PyTorchで、Tensor配列の非ゼロ要素のインデックスを取得するにはtorch.nonzeroを使用する。torch.nonzero - PyTorch Docs v1.12torch.nonzeroの使い方torch.nonzero...