Swin UnetR dose predictor