feat(torch): allow data aug to be only noise or distort

jolibrain · Jan 13, 2023 · 5a02234 · 5a02234
1 parent 5bba045
commit 5a02234
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
@@ -667,7 +667,8 @@ namespace dd
         bool has_data_augmentation
             = ad_mllib.has("mirror") || ad_mllib.has("rotate")
               || ad_mllib.has("crop_size") || ad_mllib.has("cutout")
-              || ad_mllib.has("geometry");
+              || ad_mllib.has("geometry") || ad_mllib.has("noise")
+              || ad_mllib.has("distort");
         if (has_data_augmentation)
           {
             bool has_mirror