While fine tuning the Swin UNETR, Training loss is not decreasing and training gets crashed after 10 epohs

While fine tuning the Swin UNETR, Training loss is not decreasing and training gets crashed after 10 epohs #515