diff --git a/trains/backend_interface/task/args.py b/trains/backend_interface/task/args.py index 63f639b1..f9447580 100644 --- a/trains/backend_interface/task/args.py +++ b/trains/backend_interface/task/args.py @@ -268,7 +268,8 @@ class _Arguments(object): v_type = type(v) # assume more general purpose type int -> float if v_type == int: - v_type = float + if int(v) != float(v): + v_type = float elif v_type == bool: # cast based on string or int try: diff --git a/trains/utilities/absl_bind.py b/trains/utilities/absl_bind.py index 48ef1337..ab0c0aad 100644 --- a/trains/utilities/absl_bind.py +++ b/trains/utilities/absl_bind.py @@ -70,7 +70,8 @@ class PatchAbsl(object): # noinspection PyBroadException try: if running_remotely(): - param_dict = cls._task._arguments.copy_to_dict({}, prefix=_Arguments._prefix_tf_defines) + param_dict = dict((k, FLAGS[k].value) for k in FLAGS) + param_dict = cls._task._arguments.copy_to_dict(param_dict, prefix=_Arguments._prefix_tf_defines) for k, v in param_dict.items(): # noinspection PyBroadException try: