Function get_layer_param

PATH/helper/param_count.py:72–73

(model)

#     return sum(counts)

def get_layer_param(model):
    return sum([reduce(operator.mul, i.size(), 1) for i in model.parameters()])

def measure_model(model, input_image_size, forward_param=None):
    flop_counts = []