Imported from https://github.com/wkcn/UGATIT-mxnet