tensorflow / tensorflow/models

Unclear about mobilenet v3 training config

Open
#9,852 1 comment 0 reactions 1 assignee View on GitHub

Nobody has claimed this yet.

models:research type:docs
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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.