search

HealthML / self-supervised-3d-tasks

Apache License 2.0
185 stars 39 forks source link

train the baseline model without any pre-train weights #12

Closed szhang963 closed 3 years ago

szhang963 commented 3 years ago

Hi, I do not find the code for training the baseline model without any pre-train weights. I think the code finetune.py can be modified to satisfy the need, but I don't study the tensorflow2 thoroughly. Could you give me some suggestions about the code modification? Thanks in advance.

aihamtaleb commented 3 years ago

Hi, I kindly refer you to the readme documentation: https://github.com/HealthML/self-supervised-3d-tasks/blob/master/README.md there is a flag in the finetuning config json file that is called load_weights, please try to set it to false. Best,

szhang963 commented 3 years ago

Thanks, I will read the message in detail.