Skip to content

Instantly share code, notes, and snippets.

name: "Deeplab_V2_resnet100"
input: "data"
input_dim: 1
input_dim: 3
input_dim: 321
input_dim: 321
input: "data_res05"
@jaypatravali
jaypatravali / model_params.py
Created June 9, 2017 09:13
print no of params in model
# get the number of model parameters
print('Number of model parameters: {}'.format(
sum([p.data.nelement() for p in model.parameters()])))