Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Incompatible with latest version of tensor flow etc #6

Closed
desmcm opened this issue May 15, 2020 · 10 comments
Closed

Incompatible with latest version of tensor flow etc #6

desmcm opened this issue May 15, 2020 · 10 comments

Comments

@desmcm
Copy link

desmcm commented May 15, 2020

Looks like a great package.
Unfortunately it does run on the latest version of tensor flow as various changes
have been implemented to the package.
I tried hard but could not get it to run.
Please comment here if anyone has been more successful.
Thanks, Des

@nekitmm
Copy link
Owner

nekitmm commented May 21, 2020

TF 2.0 is completely different, you need to install 1.4.0 or similar.

I am thinking of updating it actually, but never got time to do that so far... I need at some point though

@supernovae
Copy link

I was looking into effort to do this since i'd like to try the DIrectML for Windows to see if DX cards can offload some startnet++ in PixInsight. Have you had any time to look into 2.x yet?

@nekitmm
Copy link
Owner

nekitmm commented Jul 14, 2020

Working on it, but currently stuck at the step of transferring pre-trained weights to the new version

@desmcm
Copy link
Author

desmcm commented Jul 17, 2020 via email

@themissingcow
Copy link

@nekitmm, first of all, many thanks for making startnet available. It's invaluable! I don't know much about tensorflow, but I write c++/python for a living, so please let me know if I can help with any of the other aspects of the update at all.

@nekitmm
Copy link
Owner

nekitmm commented Jul 31, 2020

@nekitmm, first of all, many thanks for making startnet available. It's invaluable! I don't know much about tensorflow, but I write c++/python for a living, so please let me know if I can help with any of the other aspects of the update at all.

Thanks! I am ok on python code and everything - already wrote the whole thing. The problem is specifically with TF - old weights output complete garbage in the new version. Seems that Conv2D in TF 2.x is somehow different and I can not find out what is the problem.

@themissingcow
Copy link

The problem is specifically with TF - old weights output complete garbage in the new version. Seems that Conv2D in TF 2.x is somehow different and I can not find out what is the problem.

I asked a friend of mine who has done quite a bit of TF work, and he said:

You might need to store the weights in an external format for transfer, if you can store the network from 1.4 in Keras or ONNX format, you'll be able to read that cleanly in tf 2.0

Don't know if that helps at all?

@nekitmm
Copy link
Owner

nekitmm commented Aug 17, 2020

I finally pushed a new version made in TF2. You can check it out!

@nekitmm nekitmm closed this as completed Aug 17, 2020
@themissingcow
Copy link

Amazing, thanks so much @nekitmm.

@desmcm
Copy link
Author

desmcm commented Aug 18, 2020 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants