Imported from https://github.com/Dawars/auto_deeplab-pytorch