# the parameters are a list of [weights, biases]filters = net.params['conv1'][0].datavis_square(filters.transpose(0, 2, 3, 1))
for layer_name, param in net.params.iteritems():
# the parameters are a list of [weights, biases]
filters = net.params[layer_name][0].data
vis_square(filters.transpose(0, 2, 3, 1))
--------------------------------------------------------------------------- ValueError Traceback (most recent call last) <ipython-input-16-cf7d5999a45c> in <module>() 2 # the parameters are a list of [weights, biases] 3 filters = net.params[layer_name][0].data ----> 4 vis_square(filters.transpose(0, 2, 3, 1)) ValueError: axes don't match array
filters.transpose(0, 2, 3, 1)
filters.transpose(0, 2, 3, 1) the filters dimensions are [n_out, filter_height, filter_width, n_in].