Convolutional Autoencoder with SetNet in PyTorch - View it on GitHub