Do not bound the model width

Fastcode · Sep 4, 2020 · f72fe32 · f72fe32
1 parent c254432
commit f72fe32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/export.py b/training/export.py
@@ -91,4 +91,4 @@ def export(config, output_path):
     stages[0][0]["weights"] = first.numpy().tolist()
 
     with open(os.path.join(output_path, "model.yaml"), "w") as out:
-        yaml.dump(stages, out, default_flow_style=None, width=1000000)
+        yaml.dump(stages, out, default_flow_style=None, width=float("inf"))