Skip to content

Commit

Permalink
update doc
Browse files Browse the repository at this point in the history
  • Loading branch information
fsx950223 committed Oct 8, 2022
1 parent a0b82b2 commit 5333731
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion efficientdet/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -335,7 +335,7 @@ If you want to do inference for custom data, you can run

You should check more details of runmode which is written in caption-4.

## 9. Train on multi GPUs.
## 9. Training on single node GPUs.

Create a config file for the PASCAL VOC dataset called voc_config.yaml and put this in it.

Expand Down
4 changes: 2 additions & 2 deletions efficientdet/tf2/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -260,11 +260,11 @@ Finetune needs to use --pretrained_ckpt.

If you want to continue to train the model, simply re-run the above command because the `num_epochs` is a maximum number of epochs. For example, to reproduce the result of efficientdet-d0, set `--num_epochs=300` then run the command multiple times until the training is finished.

## 9. Train on multi GPUs.
## 9. Training on single node GPUs.

Just add ```--strategy=gpus```

## 10. Train on multi node GPUs.
## 10. Training on multi node GPUs.
Following scripts will start a training task with 2 nodes.

Start Chief training node.
Expand Down

0 comments on commit 5333731

Please sign in to comment.