Skip to content
Snippets Groups Projects

Transformer encoder

Closed Martyna Wiącek requested to merge transformer_encoder into main
Compare and
8 files
+ 463
109
Compare changes
  • Side-by-side
  • Inline
Files
8
@@ -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()}
elif isinstance(value, int) or isinstance(value, float) or isinstance(value, str):
return value
elif value is None:
return None
else:
return str(value)
@@ -145,6 +147,8 @@ class FromParameters:
parameters_dict = {}
for pn, param_value in parameters_to_serialize.items():
if pn == 'seq_encoder':
a = 0
if pn in pass_down_parameter_names:
continue
parameters_dict[pn] = serialize_single_value(param_value,