heavy augmentation, refined encoder and decoder