visualize_model(model_ft)