fine_tuned_model.fit(X_task, y_task, epochs=5, batch_size=16, validation_split=0.1)

fine_tuned_model.save('path/to/save/fine_tuned_model')

        __
        __