visualize_model(model_conv)

plt.ioff()
plt.show()