# Doubt in command Unit 3: Train your own TensorFlow

**URL:** <https://get-help.theconstruct.ai/t/doubt-in-command-unit-3-train-your-own-tensorflow/3003>\
**Category:** ROS Deep Learning With TensorFlow 101\
**Created:** [April 9, 2020, 5:45pm UTC](https://get-help.theconstruct.ai/t/doubt-in-command-unit-3-train-your-own-tensorflow/3003 "2020-04-09T17:45:43Z")\
**Posts on this page:** 3\
**Page:** 1

<div class="post-metadata">

**Author:** ![jalberola](https://get-help.theconstruct.ai/user_avatar/get-help.theconstruct.ai/jalberola/32/10_2.png) [@jalberola](https://get-help.theconstruct.ai/u/jalberola)\
**Post date:** [April 9, 2020, 5:45pm UTC](https://get-help.theconstruct.ai/t/doubt-in-command-unit-3-train-your-own-tensorflow/3003/1 "2020-04-09T17:45:43Z")

</div>

Hello,

I’m following the Unit 3 and in Step 5 we call the train.py script with some flags:

- dataset and val\_dataset: are both new files
- classes and num\_classes: correpond to the new classes

But I don’t understand what’s the point of “weights” and “weights\_num\_classes” parameters. As far as I know, the “yolov3.tf” corresponds to another previous example, and the “num\_coco\_classes” is 80.

I don’t understand these two parameters. If you can give me some explaination I would be happy.

Thanks  
Juan

---

<div class="post-metadata">

**Author:** ![duckfrost2](https://get-help.theconstruct.ai/user_avatar/get-help.theconstruct.ai/duckfrost2/32/9365_2.png) [@duckfrost2](https://get-help.theconstruct.ai/u/duckfrost2)\
**Post date:** [April 13, 2020, 9:02am UTC](https://get-help.theconstruct.ai/t/doubt-in-command-unit-3-train-your-own-tensorflow/3003/2 "2020-04-13T09:02:57Z")

</div>

Hi,

We are using the previous example DeepLearning Model Yolo with 80 clases as base for us to afterwards retrain it for only One or two clases ( in thi scase mir\_robot and objects ). So we need that data to start to be able to retrain the model.

---

<div class="post-metadata">

**Author:** ![jalberola](https://get-help.theconstruct.ai/user_avatar/get-help.theconstruct.ai/jalberola/32/10_2.png) [@jalberola](https://get-help.theconstruct.ai/u/jalberola)\
**Post date:** [April 23, 2020, 2:00pm UTC](https://get-help.theconstruct.ai/t/doubt-in-command-unit-3-train-your-own-tensorflow/3003/3 "2020-04-23T14:00:17Z")

</div>

Hi duckfrost,

Thank you for your answer.

Juan