Skip to content

Commit

Permalink
updated docs
Browse files Browse the repository at this point in the history
  • Loading branch information
@klemen1999
klemen1999 committed Sep 22, 2024
1 parent 9869dee commit 0786779
Showing 1 changed file with 19 additions and 14 deletions.
33 changes: 19 additions & 14 deletions luxonis_train/nodes/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -74,13 +74,15 @@ Adapted from [here](https://arxiv.org/pdf/2209.02976.pdf).

**Params**

| Key | Type | Default value | Description |
| ------------- | ----------- | --------------------------- | --------------------------------------------------- |
| channels_list | List\[int\] | \[64, 128, 256, 512, 1024\] | List of number of channels for each block |
| n_repeats | List\[int\] | \[1, 6, 12, 18, 6\] | List of number of repeats of RepVGGBlock |
| in_channels | int | 3 | Number of input channels, should be 3 in most cases |
| depth_mul | int | 0.33 | Depth multiplier |
| width_mul | int | 0.25 | Width multiplier |
| Key | Type | Default value | Description |
| ------------- | ----------------------------------------------------------------- | --------------------------- | --------------------------------------------------------------- |
| variant | Literal\["n", "nano", "s", "small", "m", "medium", "l", "large"\] | "nano" | Variant of the network |
| channels_list | List\[int\] | \[64, 128, 256, 512, 1024\] | List of number of channels for each block |
| n_repeats | List\[int\] | \[1, 6, 12, 18, 6\] | List of number of repeats of RepVGGBlock |
| depth_mul | float | 0.33 | Depth multiplier |
| width_mul | float | 0.25 | Width multiplier |
| block | Literal\["RepBlock", "CSPStackRepBlock"\] | "RepBlock" | Base block used |
| csp_e | float | 0.5 | Factor for intermediate channels when block=="CSPStackRepBlock" |

## RexNetV1_lite

Expand Down Expand Up @@ -143,13 +145,16 @@ Adapted from [here](https://arxiv.org/pdf/2209.02976.pdf).

**Params**

| Key | Type | Default value | Description |
| ------------- | ---------------- | ------------------------------------------------------- | ----------------------------------------- |
| n_heads | Literal\[2,3,4\] | 3 ***Note:** Should be same also on head in most cases* | Number of output heads |
| channels_list | List\[int\] | \[256, 128, 128, 256, 256, 512\] | List of number of channels for each block |
| n_repeats | List\[int\] | \[12, 12, 12, 12\] | List of number of repeats of RepVGGBlock |
| depth_mul | int | 0.33 | Depth multiplier |
| width_mul | int | 0.25 | Width multiplier |
| Key | Type | Default value | Description |
| ------------- | ----------------------------------------------------------------- | ------------------------------------------------------- | --------------------------------------------------------------- |
| variant | Literal\["n", "nano", "s", "small", "m", "medium", "l", "large"\] | "nano" | Variant of the network |
| n_heads | Literal\[2,3,4\] | 3 ***Note:** Should be same also on head in most cases* | Number of output heads |
| channels_list | List\[int\] | \[256, 128, 128, 256, 256, 512\] | List of number of channels for each block |
| n_repeats | List\[int\] | \[12, 12, 12, 12\] | List of number of repeats of RepVGGBlock |
| depth_mul | float | 0.33 | Depth multiplier |
| width_mul | float | 0.25 | Width multiplier |
| block | Literal\["RepBlock", "CSPStackRepBlock"\] | "RepBlock" | Base block used |
| csp_e | float | 0.5 | Factor for intermediate channels when block=="CSPStackRepBlock" |

## ClassificationHead

Expand Down

0 comments on commit 0786779

Please sign in to comment.