Pytorchのモデル構築、評価を補助するVisdomとtorchsummary - Qiita
タイトルの通りなんですが、pytorchで地味に役立つ２つのライブラリについての説明を実際にCIFAR10の識別モデルをうごかしながら説明します。 tl;dr VisdomはTensorboardのPytorch版だよ torchsummaryはKerasでいうところのmodel.summaryだよ Visdomの使い方の例を実際のモデルを動かしながら説明している記事がなかったから書いたよ class Model(nn.Module): def __init__(self, nch=3, n_classes=10): super(Model, self).__init__() self.layers = nn.ModuleList([ nn.Sequential( nn.Conv2d(nch, 32, 3, 1, 1), nn.ReLU(inplace=True), nn.MaxPool2
リリース、障害情報などのサービスのお知らせ
最新の人気エントリーの配信
j次のブックマーク
k前のブックマーク
lあとで読む
eコメント一覧を開く
oページを開く
A Multitask Music Model with BERT, Transformer-XL and Seq2Seq
Google QUEST Q&A Labeling