try different loss. change encoder pretrained augmentation