-
Kai Zhang authored
Summary: Currently we are checking if MODEL.DEVICE is "gpu", but actually we DEVICE could also be "cuda". This diff checks if device is "cpu" instead. Reviewed By: wat3rBro Differential Revision: D28689547 fbshipit-source-id: 7512d32b7c08b0dcdc6487c6c2f1703655e64b19
bf395ce5
