hsl

1 files changed, 6 insertions, 6 deletions

diff --git a/options/dataset_options.py b/options/dataset_options.py
index 8b638e0..1692766 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -284,22 +284,22 @@ class DatasetOptions(BaseOptions):
 
         self.parser.add_argument(
           '--hue',
-          default=0.0,
-          type=float,
+          default=0,
+          type=int,
           help='hue rotation'
         )
 
         self.parser.add_argument(
           '--saturation',
-          default=0.0,
-          type=float,
+          default=0,
+          type=int,
           help='saturation'
         )
 
         self.parser.add_argument(
           '--luminosity',
-          default=0.0,
-          type=float,
+          default=0,
+          type=int,
           help='luminosity'
         )