Fix different env variables in schedules runs pt 2 (#15934)
Summary:
Unfortunately I do not know how to test this without merging it first
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15934
Reviewed By: orionr
Differential Revision: D13627472
Pulled By: pjh5
fbshipit-source-id: 35eced1483bbf3c0c3f6f62fb7bbbf2f200e50e6