Skip to content
Snippets Groups Projects

Transformer encoder

8 files
+ 463
109
Compare changes
  • Side-by-side
  • Inline

Files

+ 4
0
@@ -58,6 +58,8 @@ def serialize_single_value(value: Any, pass_down_parameter_names: List[str] = No
@@ -58,6 +58,8 @@ def serialize_single_value(value: Any, pass_down_parameter_names: List[str] = No
return {k: serialize_single_value(v, pass_down_parameter_names) for k, v in value.items()}
return {k: serialize_single_value(v, pass_down_parameter_names) for k, v in value.items()}
elif isinstance(value, int) or isinstance(value, float) or isinstance(value, str):
elif isinstance(value, int) or isinstance(value, float) or isinstance(value, str):
return value
return value
 
elif value is None:
 
return None
else:
else:
return str(value)
return str(value)
@@ -145,6 +147,8 @@ class FromParameters:
@@ -145,6 +147,8 @@ class FromParameters:
parameters_dict = {}
parameters_dict = {}
for pn, param_value in parameters_to_serialize.items():
for pn, param_value in parameters_to_serialize.items():
 
if pn == 'seq_encoder':
 
a = 0
if pn in pass_down_parameter_names:
if pn in pass_down_parameter_names:
continue
continue
parameters_dict[pn] = serialize_single_value(param_value,
parameters_dict[pn] = serialize_single_value(param_value,
Loading