diff --git a/classification/AntBee/config/train_test_ce1.cfg b/classification/AntBee/config/train_test_ce1.cfg
index 7c8e008..5ce844f 100755
--- a/classification/AntBee/config/train_test_ce1.cfg
+++ b/classification/AntBee/config/train_test_ce1.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
diff --git a/classification/AntBee/config/train_test_ce2.cfg b/classification/AntBee/config/train_test_ce2.cfg
index 4a2f954..b1c8795 100755
--- a/classification/AntBee/config/train_test_ce2.cfg
+++ b/classification/AntBee/config/train_test_ce2.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
diff --git a/classification/CHNCXR/config/net_resnet18.cfg b/classification/CHNCXR/config/net_resnet18.cfg
index df8029f..26ade74 100755
--- a/classification/CHNCXR/config/net_resnet18.cfg
+++ b/classification/CHNCXR/config/net_resnet18.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
diff --git a/classification/CHNCXR/config/net_vgg16.cfg b/classification/CHNCXR/config/net_vgg16.cfg
index 35ec1fe..a9fd095 100755
--- a/classification/CHNCXR/config/net_vgg16.cfg
+++ b/classification/CHNCXR/config/net_vgg16.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]