Imported from https://github.com/DeadAt0m/DCA-PyTorch