Skip to content

Commit

Permalink
Fix documentation (openvinotoolkit#56)
Browse files Browse the repository at this point in the history
  • Loading branch information
vshampor committed Jul 20, 2020
1 parent a0c1c2b commit 0df773d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion docs/Usage.md
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,8 @@ The framework is designed so that the modifications to your original training co
```
The `create_compressed_model` function parses the loaded configuration file and returns two objects. `compression_ctrl` is a "controller" object that can be used during compressed model training to adjust certain parameters of the compression algorithm (according to a scheduler, for instance), or to gather statistics related to your compression algorithm (such as the current level of sparsity in your model).

5. (Optional) Wrap your model with `DataParallel` or `DistributedDataParallel` classes for multi-GPU training. If you do so, add the following call afterwards:
5. (Optional) Wrap your model with `DataParallel` or `DistributedDataParallel` classes for multi-GPU training.
If you use `DistributedDataParallel`, add the following call afterwards:
```python
compression_ctrl.distributed()
```
Expand Down

0 comments on commit 0df773d

Please sign in to comment.