search

Raukk / tf-keras-surgeon

Pruning and other network surgery for trained TF.Keras models.
Other
32 stars 9 forks source link

Install instructions and examples incorrect #4

Open nateGeorge opened 5 years ago

nateGeorge commented 5 years ago

I think your install instructions are incorrect. You say pip install kerassurgeon but it should be pip install tfkerassurgeon. Also, the examples show import kerassurgeon instead of import tfkerassurgeon. The colab example seems to have it correct.

Raukk commented 4 years ago

Sorry for the delay.

I've updated the Readme to be more clear.

The bottom half is unedited from the Keras-Surgeon project. I intended to rewrite all the Readme, docs, etc. but I have not had as much time as I'd like (and I have too many projects I want to do).

If you would like to contribute edits to the docs/code, please feel free to open a pull request.

If you have any more feedback or suggestions, please let me know! Thank you