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

Could you public the saved model for tranfer learning? #8

Closed
ashishgupta1350 opened this issue Apr 18, 2020 · 1 comment
Closed

Could you public the saved model for tranfer learning? #8

ashishgupta1350 opened this issue Apr 18, 2020 · 1 comment

Comments

@ashishgupta1350
Copy link

Hi,
Really loved your work, I went through your code and through the paper, the code is well written and documented.
I tried running the model and yes, it does take a lot of time to run on CPU, especially when I try to run on a mobile chip of a laptop. Unfortunately I don't have the luxury to get a GPU because of 2 reasons:

  1. The labs in our departments are closed due to quarentine for another 1.5 months
  2. I don't have a pre - existing GPU to run CUDA .

If you could make your transfer learning model public ( Ie the trained model created from model.save() ) It would be a great help. I could then alter your code and use transfer learning for my usecase.

1 solution I found was to use 100K data points to train the initial network ( LSTM Cell + RNN ) , however that is also quiet time consuming surprisingly on the CPU + the results will suffer.

Awaiting your reply
Your's Sincerely

@topazape
Copy link
Owner

Hi, @ashishgupta1350

Thanks for your comment.
As you mentioned, to fine-tune, you have to retrain from the model ./experiments/2020-03-24/LSTM_Chem/checkpoints/LSTM_Chem-22-0.45.hdf5. (If you don't know how to retrain, look at ./example_Fine-tuning_for_TRPM8.ipynb)Of course, the more GPU you have, the faster you can retrain.

So I would recommend using google colaboratory. You can use the GPU for free on it. However, I have not tested this repository to work properly on google colaboratory.

Sincerely,

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

2 participants