ha-ha-ha-han / UKBiobank_deep_pretrain

Pretrained neural networks for UK Biobank brain MRI images. SFCN, 3D-ResNet etc.
MIT License
129 stars 36 forks source link

Training? #4

Closed willianpripp closed 3 years ago

willianpripp commented 3 years ago

Hello, I am trying to replicate your SFCN brain age prediction model to do my undergraduate thesis on top of it. My idea is to use your model that has great results and retrain it with part of my data and your input data. But I'm having some problems with the code, because machine learning is kind of new to me, could you share the code that was used to train the model or part of it, some parameters, anything would help? Thanks.

ha-ha-ha-han commented 3 years ago

Yes. Nearly there - I'll release a tutorial notebook in a week or two.

But I'm having some problems with the code

Can you locate the bug (if any)?

willianpripp commented 3 years ago

Thanks, I'll be waiting for the tutorial, apreciate a lot.

willianpripp commented 3 years ago

Sorry to bother, but do you have some feedback for the tutorial?