I would like to run stage_two/demo.py but i thing i need "epoch_1_step_9378.pth" file where can i find it ?