K-Means Init Error

[phongdinhv]

I would like to add more layers to the Network and init it using the magic_init_mod.py. When i tried to run the following command:

python resources/magic_init/magic_init_mod.py 'path/models/colorization_train_val_v2_custome_30.prototxt' 30_layers.caffemodel -d 'img/ILSVRC2012_test_00005167.JPEG' -q -nit 10 -cs

Everything went right until the last line showed an error:

Traceback (most recent call last):
File "resources/magic_init/magic_init_mod.py", line 511, in
main()
File "resources/magic_init/magic_init_mod.py", line 507, in main
calibrateGradientRatio(n)
File "resources/magic_init/magic_init_mod.py", line 312, in calibrateGradientRatio
output_std = np.mean(np.std(flattenData(active_data[last_tops[0]]), axis=0))
File "resources/magic_init/magic_init_mod.py", line 29, in flattenData
return np.concatenate([d.swapaxes(0, 1).reshape((d.shape[1],-1)) for d in data], axis=1).T
ValueError: bad axis1 argument to swapaxes

I've also tried the default model which is "colorization_train_val_v2.prototxt'" but still had the same issue