soumik12345 / MIRNet

Tensorflow implementation of MIRNet for Low-light image enhancement
https://huggingface.co/spaces/keras-io/Enhance_Low_Light_Image
Apache License 2.0
112 stars 37 forks source link

Not able to save the model after training on local dataset. #8

Open AbdulRahmaan03 opened 7 months ago

AbdulRahmaan03 commented 7 months ago

Tried saving with .h5 as well as the .pb extension.

`---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[21], line 1
----> 1 model.save('mirnet.h5')

File ~\anaconda3\envs\tf\lib\site-packages\keras\utils\traceback_utils.py:70, in filter_traceback.<locals>.error_handler(*args, **kwargs)
     67     filtered_tb = _process_traceback_frames(e.__traceback__)
     68     # To get the full stack trace, call:
     69     # `tf.debugging.disable_traceback_filtering()`
---> 70     raise e.with_traceback(filtered_tb) from None
     71 finally:
     72     del filtered_tb

File ~\anaconda3\envs\tf\lib\site-packages\keras\utils\generic_utils.py:545, in serialize_keras_object(instance)
    543     raise e
    544 serialization_config = {}
--> 545 for key, item in config.items():
    546     if isinstance(item, str):
    547         serialization_config[key] = item

AttributeError: 'NoneType' object has no attribute 'items`