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

[kenmatsu4]

“from torchsummary import summary summary(your_model, input_size=(channels, H, W))”