lg_unetr.py should be used with https://github.com/Project-MONAI/research-contributions/tree/main/SwinUNETR/BRATS21, you can subsititude it with the original Swin UNETR. The trained model is now available via Google Drive at the following link: https://drive.google.com/file/d/1tqmPZwftruTsRCn2BI2Hpkgt5BNwxMSG/view?usp=sharing