Skip to content

Commit 7df2014

Browse files
author
abdelabd
committed
updated error handling
1 parent 9850455 commit 7df2014

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/writer/vivado_writer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -662,7 +662,7 @@ def keras_model_representer(dumper, keras_model):
662662
with open(model.config.get_output_dir() + '/' + config_filename, 'w') as file:
663663
try:
664664
yaml.dump(model.config.config, file)
665-
except ValueError:
665+
except:
666666
import torch
667667
model_path = model.config.get_output_dir() + "/torch_model_state_dict.pt"
668668
torch.save(model.config.config["PytorchModel"].state_dict(), model_path)

0 commit comments

Comments
 (0)