diff --git a/cml/shared/parameter.py b/cml/shared/parameter.py
index 076b31a13a0afdfa5fd0ab3b2dfd9a42af9b6b4a..1dfaab64bbc3e3117b6ca877d9a43159048efb3b 100644
--- a/cml/shared/parameter.py
+++ b/cml/shared/parameter.py
@@ -149,7 +149,13 @@ class StackIterations:
         return self.stack_iterations
 
     def __set__(self, instance, value):
-        self.stack_iterations = value
+        if isinstance(value, int):
+            self.stack_iterations = value
+        elif isinstance(value, str):
+            try:
+               self.stack_iterations = int(value)
+            except Exception as error:
+                print(error)
 
 
 class LearnblockMinimum: