Fix export of multidimensional params (facebookresearch#214)

Summary: Pull Request resolved: facebookresearch#214 Reviewed By: adamlerer Differential Revision: D28786238 fbshipit-source-id: 3613c5ae705367895ccf4b5b2fe54f9d6bcc5b54
kevin-14 · Jun 7, 2021 · 1598839 · 1598839
1 parent b049325
commit 1598839
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchbiggraph/converters/export_to_tsv.py b/torchbiggraph/converters/export_to_tsv.py
@@ -96,7 +96,7 @@ def make_tsv_for_relation_types(
                     write(
                         relation_types_tf,
                         (rel_t_name, side, op_name, param_name, shape),
-                        param,
+                        param.flatten(),
                     )
     else:
         for rel_t_name, rel_t_config, operator in zip(
@@ -113,7 +113,7 @@ def make_tsv_for_relation_types(
                 write(
                     relation_types_tf,
                     (rel_t_name, "rhs", op_name, param_name, shape),
-                    param,
+                    param.flatten(),
                 )
 
     relation_types_output_filename = getattr(