I also visited the repository and I found it had some issues. For example, the transforms would be slower when applying nn.DataParallel even if one had more than 1 GPU. Moreover, it seems that this repo hasn’t been updated since PyTorch 0.4.0.