人工知能とか犬とか

人工知能と犬に興味があります。しょぼしょぼ更新してゆきます。

DeepLearning

PyTorchのDatasetとDataLoader

概要 PyTorchのチュートリアルData Loading and Processing Tutorial をやってみて、DatasetとDataLoaderの使い方を学ぶのです。 概要 DatasetとDataLoader Dataset DataLoader TransformとCompose (おまけ)DataLoaderのcollate_fn まとめ DatasetとDataLo…

PyTorchの分散環境学習

概要 PyTorchのチュートリアルに、分散環境での学習に関する記事がある。 自分の家のサーバーはシングルGPUなのだが、最近少々不満を感じてきている。 クラウドでの分散学習を見越して、今後のために勉強しておくことにしたのです。 なお、特に実装はありま…

PyTorchのSeq2Seqをミニバッチ化するときに気をつけたこと

概要 PyTorchチュートリアルに、英仏の機械翻訳モデルを作成するTranslation with a Sequence to Sequence Network and Attentionがあります。 このチュートリアルは、教師データを一つずつ与える形になっており、結構遅いのです。 なので、バッチでの学習が…

PyTorchのRNNとRNNCell

概要 PyTorchでRNNを使った実装しようとするとき、torch.nn.RNNとtorch.nn.RNNCellというものがあることに気がつきました。 それぞれの違いを明らかにして、注意点を整理しておきたいのです。 リカレント層の実装方法 PyTorchチュートリアルの、名前分類をこ…

PyTorchでわんにゃん分類器をつくる

概要 PyTorchで事前学習済みモデルのファインチューニングを行って、犬や猫の種類を分類できる分類器を作成しました。使用している事前学習済みモデルはResNet18、データセットはThe Oxford-IIIT Pet Datasetを使用します。 特になんの工夫もしなくても、90%…