Hi Adam,
Yes - running the scripts in nvidia-docker with --ipc=host.
I got some strange behavior, so opened an issue about it here.  https://github.com/pytorch/pytorch/issues/1120 52
Looks like models are hardcoded with 224x224 dimensions.  I am looking into it now.