Zhang Yimou's "Impressions of West Lake"

Hi there,
I intend to catch Zhang Yimou's "Impression of West Lake" show and still hope to catch the last train back to Shanghai. Is this possible or do I need to stay over. My accommodation in Shanghai is already confirmed. Thanks.