zhanghang1989 / zhanghang1989/PyTorch-Encoding
About parameters --syncbn and --ngpus in the train.py
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 2k
- Forks
- 448
- PR merge metrics
- No merged PRs in 30d
Description
Hi,when I use the train.py to train EncNet_ResNet101s_VOC on the Pascal Voc dataset,I click the cmd link to get command。The command as follows:
~#First finetuning COCO dataset pretrained model on augmented set
~# You can also train from scratch on COCO by yourself
CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py --dataset Pascal_aug --model-zoo EncNet_Resnet101_COCO --aux --se-loss --lr 0.001 --syncbn --ngpus 4 --checkname res101 --ft
~# Finetuning on original set
CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py --dataset Pascal_voc --model encnet --aux --se-loss --backbone resnet101 --lr 0.0001 --syncbn --ngpus 4 --checkname res101 --resume runs/Pascal_aug/encnet/res101/checkpoint.params --ft
when I use the first command, the system point out no parameters about --syncbn and --ngpus。just to delete the --syncbn and --ngpus ,the command can work。Meanwhile, I can not find these two parameters in the train.py。
So, are these two parameters necessary? In other words, do I need to modify the contents of the train.py files to use them? Or is it just because the cmd link content has not been updated? I just need to simply delete these two parameters?
Thank you! Looking forward to your reply。
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with train.py and compare its accepted arguments with the two commands in the issue and the linked command instructions. Done means the documented commands use supported options and clearly state whether --syncbn and --ngpus should be retained or removed.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- cli, machine-learning
- Issue type
- Documentation
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100