Resumo do modelo em pytorch
Existe alguma maneira, eu posso imprimir o resumo de um modelo em PyTorch como o model.summary()método faz em Keras como segue? Model Summary: ____________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ==================================================================================================== input_1 (InputLayer) (None, 1, 15, 27) 0 ____________________________________________________________________________________________________ convolution2d_1 (Convolution2D) (None, 8, 15, 27) 872 input_1[0][0] ____________________________________________________________________________________________________ …