tensorflow / tensorflow/models
Unclear about mobilenet v3 training config
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Hi, I tried to train mobilenet v3 following the configuration in page. However, some of the configurations are not defined in train_image_classifier.py:
init_stddev | 0.008 |
dropout_keep_prob | 0.8 |
bn_moving_average_decay | 0.997 |
bn_epsilon | 0.001 |
Even though I tried to define these hyperparameters, I couldn't find where to define them. For example, I can't find where to define dropout_keep_prob, as the mobilenetv3 network doesn't include a dropout layer and the slim.conv2d layer has not a 'dropout' argument. Could you please further explain how to do these configurations? Or could you kindly provide the training scripts?
In addition, how do you train mobilenetv3 on multiple GPUs? I think train_image_classifier.py only supports single GPU training.
Contributor guide
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.
Assessment
This issue has not been assessed yet.