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

Problem with pseudo-label generation #7

Open
Rzy1221 opened this issue Nov 27, 2023 · 2 comments
Open

Problem with pseudo-label generation #7

Rzy1221 opened this issue Nov 27, 2023 · 2 comments

Comments

@Rzy1221
Copy link

Rzy1221 commented Nov 27, 2023

Dear author, when I run this command,Create folder root/data/pseudo_labels/5shots/ and generate pseudo labels by running:

CUDA_VISIBLE_DEVICES=0 python tools/train_mt_part.py --cfg experiments/ap10k/hrnet/w32_256x192_adam_lr1e-3.yaml --animalpose --generate_pseudol --pretrained output/output_animal_hrnet5_part/model_best.pth OUTPUT_DIR test MODEL.NAME pose_hrnet_part TRAIN.BATCH_SIZE_PER_GPU 32 GPUS [0,] LABEL_PER_CLASS 5

There's been an error. My CPU usage is high and seems to be stuck
image

My first step of training with 5 images per species went well, but I got stuck generating false labels.
Looking forward to your answer.

@chaneyddtt
Copy link
Owner

Hi @Rzy1221, it is not stuck, it is generating pseudo labels and the process is quite slow.

@Rzy1221
Copy link
Author

Rzy1221 commented Nov 27, 2023

Thanks for your reply! I got the result in a few longer hours.
image

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