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

Do you have a keras version? #1

Open
xuzhang5788 opened this issue May 2, 2018 · 12 comments
Open

Do you have a keras version? #1

xuzhang5788 opened this issue May 2, 2018 · 12 comments
Labels
help wanted Extra attention is needed

Comments

@xuzhang5788
Copy link

No description provided.

@izmailovpavel
Copy link
Collaborator

Hi, currently we don't have any keras implementation or examples. We will try to provide an example some time soon, probably after may 18. If you come up with an example, please post it in this issue.

@xuzhang5788
Copy link
Author

Thank you so much. I am looking forward to seeing your new implementation.

@nile649
Copy link

nile649 commented May 25, 2018

Hi, so where you guys able to make keras or tensorflow implementation? I did try your pytorch version code with cyclic learning rate. looking forward for keras implementation, if possible just provide the function.

Thanks

@izmailovpavel izmailovpavel added the help wanted Extra attention is needed label Aug 7, 2018
@happysmile123
Copy link

Hi, I use keras+tensorflow build a swa model ,the dataset is mnist ,i tried with cycle lr ,but i found the result not so stable ,sometimes the swa is better than single model ,however sometimes not .And I saw you trained 200-300epoches, i just trained several epoch like 2 or 3,but the accuracy is 0.98 or even better ,so when i averaged the weight , there seems to be little improvement .so I don not know whether swa is worth to use . After all ,if you could provide a keras or tensorflow version ,it could be better

@xuzhang5788
Copy link
Author

Here is a SWA keras version (https://github.com/kristpapadopoulos/keras-stochastic-weight-averaging). But it only can do a constant learning rate for a user defined amount of epochs.

His swa.py is very similar to another swa.py showed in the article "Adding a cutting-edge deep learning training technique to the fast.ai library" at Medium.

@izmailovpavel Do you think we can use this keras implementation in our keras model training? Many thanks.

@samjia533332
Copy link

samjia533332 commented Nov 4, 2018 via email

@xuzhang5788
Copy link
Author

@samjia533332
Would you like to give us a link about your found version? Many thanks.

@Venka97
Copy link

Venka97 commented Dec 28, 2018

@samjia533332 @xuzhang5788 Did you find the version with cyclic lr for keras?

@simon-larsson
Copy link

I have implemented the constant and cyclic lr schedules with the batch norm fix in Keras here: https://github.com/simon-larsson/keras-swa

I´m not 100% on the cyclic schedule since there is no implementation of it here that I could use as reference.

@izmailovpavel
Copy link
Collaborator

@simon-larsson that's great :) Do the results you are getting look similar to what we get in PyTorch?

@simon-larsson
Copy link

@izmailovpavel the constant learning rate schedule gave similar results on cifar10 and 2-3% acc on some private datasets. I want to replicate more when I get access to hardware again.

Is it ok if I email you some questions about the cyclic learning schedule?

@izmailovpavel
Copy link
Collaborator

@simon-larsson great! Sure, feel free to email me at pi390@nyu.edu.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

7 participants