Update requirements.txt and there is no json file outputed

vbelz / Speech-enhancement

Deep learning for audio denoising

MIT License

620 stars 124 forks source link

Update requirements.txt and there is no json file outputed #21

Open b-aaz opened 2 years ago

b-aaz commented 2 years ago

Hello there! So I had a journey installing all dependencys and python it would be great if you update the requirements.txt file! Iam guessing that my problem is cussed by the worng version of tensorflow. But my problem is that after training there won't be any .json file outputed in any folder It would be great if you help me 😃

lay123design commented 2 years ago

"my problem is that after training there won't be any .json file outputed in any folder",so am I. why?

neng668 commented 2 years ago

You can add from line 66 of train_model.py:

model_json = generator_nn.to_json()
with open(("model_best/json", "w") as json_file:
    json_file.write(model_json)

You will also need to downgrade h5py to 2.10.0 otherwise you may get some decoding errors.