U-Net structure compatible with 2D data and 3D data, and different loss functions are implemented. Currently supported loss functions focal loss dice loss cross entropy loss distance loss edge loss Currently supported network structures U-Net V-Net U-Net++ attention U-Net deeplab v1-3 support multi-GPU training, synchronized batch normalization