Fixed deprecated use of yaml.load
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/22985
Test Plan: Imported from OSS
Differential Revision: D16425112
Pulled By: Chillee
fbshipit-source-id: ef0c764c3fd2518b9284d9a20e84d677ebd8f277