I converted the pre-trained model to PyTorch. The PyTorch model uses the standard torchvision ResNet implementation and can be loaded without any other imports. Find the PyTorch model under training/ae_model_pytorch.pt.
Original paper and code is by Katharina Schwarz Patrick Wieschollek, Hendrik P.A. Lensch