Skip to content

Commit f72fe32

Browse files
Do not bound the model width
1 parent c254432 commit f72fe32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

training/export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,4 +91,4 @@ def export(config, output_path):
9191
stages[0][0]["weights"] = first.numpy().tolist()
9292

9393
with open(os.path.join(output_path, "model.yaml"), "w") as out:
94-
yaml.dump(stages, out, default_flow_style=None, width=1000000)
94+
yaml.dump(stages, out, default_flow_style=None, width=float("inf"))

0 commit comments

Comments
 (0)