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

local-rank #26

Open
TasosTzaras opened this issue Jul 30, 2024 · 0 comments
Open

local-rank #26

TasosTzaras opened this issue Jul 30, 2024 · 0 comments

Comments

@TasosTzaras
Copy link

i run the training example as it is and i get this error
Note that --use-env is set by default in torchrun.
If your script expects --local-rank argument to be set, please
change it to read from os.environ['LOCAL_RANK'] instead.
usage: ssda_yolov5_train.py [-h] [--weights WEIGHTS] [--cfg CFG] [--data DATA] [--hyp HYP]
[--epochs EPOCHS] [--batch-size BATCH_SIZE]
[--img-size IMG_SIZE [IMG_SIZE ...]] [--rect] [--resume [RESUME]]
[--nosave] [--notest] [--noautoanchor] [--evolve] [--bucket BUCKET]
[--cache-images] [--image-weights] [--device DEVICE] [--multi-scale]
[--single-cls] [--adam] [--sync-bn] [--workers WORKERS]
[--project PROJECT] [--entity ENTITY] [--name NAME] [--exist-ok]
[--quad] [--linear-lr] [--label-smoothing LABEL_SMOOTHING]
[--upload_dataset] [--bbox_interval BBOX_INTERVAL]
[--save_period SAVE_PERIOD] [--artifact_alias ARTIFACT_ALIAS]
[--local_rank LOCAL_RANK] [--teacher_alpha TEACHER_ALPHA]
[--conf_thres CONF_THRES] [--iou_thres IOU_THRES]
[--max_gt_boxes MAX_GT_BOXES] [--lambda_weight LAMBDA_WEIGHT]
[--consistency_loss] [--alpha_weight ALPHA_WEIGHT]
[--student_weight STUDENT_WEIGHT] [--teacher_weight TEACHER_WEIGHT]
[--save_dir SAVE_DIR]
ssda_yolov5_train.py: error: unrecognized arguments: --local-rank=0

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

1 participant