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

how to use pretrained model? #19

Closed
StawEndl opened this issue Jan 11, 2023 · 1 comment
Closed

how to use pretrained model? #19

StawEndl opened this issue Jan 11, 2023 · 1 comment
Labels
question Further information is requested

Comments

@StawEndl
Copy link

hi, i want to use pretained model for new trainning, how should i do? i want to trainning with dataset B, and i want to use model_0159999.pth which is producted at trainning on dataset A, as the pretrained model.

@praeclarumjj3 praeclarumjj3 added the question Further information is requested label Jan 12, 2023
@praeclarumjj3
Copy link
Member

Hi, you can directly set the MODEL.WEIGHTS as model_0159999.pth to use it as the pretrained checkpoint.

python train_net.py --dist-url 'tcp://127.0.0.1:50163' \
    --num-gpus 8 \
    --config-file <path-to-config>  \
    MODEL.WEIGHTS model_0159999.pth \
    OUTPUT_DIR <output_dir> WANDB.NAME <name>

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants