Imported from https://github.com/reiinakano/neural-painters-pytorch