From 7765b7ffa674ab9feaae02f0fbc86f4769b1c745 Mon Sep 17 00:00:00 2001 From: Mohammad Alshamsi <33751441+MohammadRobot@users.noreply.github.com> Date: Thu, 29 Oct 2020 14:05:39 +0400 Subject: [PATCH] Update pytorch-ssd.md update the following arguments:- `--epochs` ==> `--num-epochs` `--workers` ==> `--num-workers` --- docs/pytorch-ssd.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/pytorch-ssd.md b/docs/pytorch-ssd.md index 4abaaff84..83dae249a 100644 --- a/docs/pytorch-ssd.md +++ b/docs/pytorch-ssd.md @@ -132,8 +132,8 @@ Here are some common options that you can run the training script with: | `--model-dir` | `models/` | directory to output the trained model checkpoints | | `--resume` | None | path to an existing checkpoint to resume training from | | `--batch-size` | 4 | try increasing depending on available memory | -| `--epochs` | 30 | up to 100 is desirable, but will increase training time | -| `--workers` | 2 | number of data loader threads (0 = disable multithreading) | +| `--num-epochs` | 30 | up to 100 is desirable, but will increase training time | +| `--num-workers`| 2 | number of data loader threads (0 = disable multithreading) | Over time, you should see the loss decreasing: