Change input shape dimensions for fine-tuning with Keras