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

Does signle 24G card is enough to finetune the model for training? #22

Open
lsq117 opened this issue Jun 21, 2024 · 2 comments
Open

Does signle 24G card is enough to finetune the model for training? #22

lsq117 opened this issue Jun 21, 2024 · 2 comments

Comments

@lsq117
Copy link

lsq117 commented Jun 21, 2024

i want to konw whether i can use my 3090ti(24G) to training the model? on the other hand, what do you think that load the CT pretrained weight to train the MRI data? hoping for your replay

@Yuxin-Du-Lab
Copy link
Collaborator

24G memory will limit the batch size seriously. I can't guarantee the quality of fine-tuning. I think CT pre-trained weight is some kind of helpful for the training on MRI data, just better than training from scratch.

@Yuxin-Du-Lab
Copy link
Collaborator

If you have multiple 3090ti(24G)s, full parameter finetuning is possible.

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