The nearest stations at which the trains halt are Castle Rock and Kulem, before and after crossing the falls respectively.
If you want to witness the falls from train, they are located just after the Dudhsagar Station. After Dudhsagar Station, the train traverses through a tunnel after which it would cross the falls.
The Braganza ghat section starts after Castle Rock and ends at Kulem.