DNN-printer 0.0.1

Print feature map and weight size of PyTorch models.

# Split_layer This is a simple package to print the size of feature map and DNN weight.

## Usage

You can use this tool by three steps:

0. Install printer by running pip3 install printer -U –user

1. Find the file which defines the structure of a Network. Add the following code:

from printer import printer

def train(epoch):

print(’nEpoch: %d’ % epoch) net.train() train_loss = 0 correct = 0 total = 0 printer(net, (3, 32, 32),batch_size) … …

Notice: The first parameter is the net, the second paramater is the size of input data and the third parameter is the batch size.

## result


## Requirements Make sure torch has been installed.