search

pythonlessons / mltu

Machine Learning Training Utilities (for TensorFlow and PyTorch)
MIT License
160 stars 100 forks source link

make Model2onnx agnostic to the type of model used by tensorflow #54

Closed seidnerj closed 1 month ago

seidnerj commented 1 month ago

Currently Model2onnx expects a file with an extension of "h5" models (HDF5 format), but really there's no reason why this would be dependant on the file extension. This makes Model2onnx agnostic to the file extension (and also the tensorflow actual model type).

This replaces PR https://github.com/pythonlessons/mltu/pull/53

pythonlessons commented 1 month ago

I merge this pull request to develop, later this will be integrated into main branch