-
Notifications
You must be signed in to change notification settings - Fork 1.3k
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
Minimal jupyter notebook to train models #15
Comments
Can you make that link public? Or maybe just upload the notebook to this issue page? |
Now the link should work. I attach the notebook too. |
Cool! Looks good! I had just released the torchxrayvision code! I have been doing something similar but I wasn't able to get good prediction performance. Can you include an AUC calculation on the resulting model? Also, I would expect monitoring test performance here as you do would cause you to overfit. Maybe leave out a few cases that you only test the final model on. |
Yes, good ideas to keep improving it. I'll do it when I have spare time during the week. |
Thanks for sharing, |
@jovenwayfarer Labels are binary. My code is to predict COVID19 status stored at column 2. Tomorrow I'll improve it with some ideas.... |
@rcillavicomtech I tested your code 4 days ago, but due to CUDA usage in the snippet I had to run on colab, because I don't have access to any Nvidia GPU, so it took quiet some time to train the model... I am also trying to find more datasets, I look forward of your improved code |
|
@ieee8023 Hi, could you help? |
You can view the code here: https://github.com/mlmed/torchxrayvision/ If you specify weights you will get pretrained DenseNet 121 models based on the models trained for this paper: https://arxiv.org/abs/2002.02497 |
@rcillavicomtech not sure to understand your training, what does
|
@rcillavicomtech Thank you for your great work! When I used the covid dataset the same way as yours, I encountered an assertion error:
Can anyone help with this problem? Thank you in advance! |
Hi,
I'm a deep learning researcher from Spain.
I have created a minimal jupyter notebook to train with the images from your repo.
https://drive.google.com/file/d/19T_qebLa1keUNpkp7FDNEBmsNRQUVtfJ/view?usp=sharing
I want to experiment with visual attention models on next days.
Rodrigo
The text was updated successfully, but these errors were encountered: