Hi, having the same issue here https://github.com/pytorch/examples/pull/47/files#diff-1357f5fd365db925427f5f0692df5b52R251