Skip to content

Commit

Permalink
Update use.md
Browse files Browse the repository at this point in the history
  • Loading branch information
NorthOcean authored Sep 25, 2023
1 parent 87c0847 commit 443caae
Showing 1 changed file with 37 additions and 0 deletions.
37 changes: 37 additions & 0 deletions use.md
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,7 @@ About the `argtype`:
The program will parse these args from the terminal at each time.

<!-- DO NOT CHANGE THIS LINE -->

### Basic args

- `--K_train`: type=`int`, argtype=`static`.
Expand All @@ -222,6 +223,12 @@ About the `argtype`:
- `--K`: type=`int`, argtype=`dynamic`.
Number of multiple generations when testing. This arg only works for multiple-generation models.
The default value is `20`.
- `--anntype`: type=`str`, argtype=`static`.
Model's predicted annotation type. Can be `'coordinate'` or `'boundingbox'`.
The default value is `coordinate`.
- `--auto_dimension`: type=`int`, argtype=`temporary`.
Choose whether to handle the dimension adaptively. It is now only used for silverballers models that are trained with annotation type `coordinate` but want to test on datasets with annotation type `boundingbox`.
The default value is `0`.
- `--batch_size` (short for `-bs`): type=`int`, argtype=`dynamic`.
Batch size when implementation.
The default value is `5000`.
Expand Down Expand Up @@ -279,9 +286,18 @@ About the `argtype`:
- `--obs_frames` (short for `-obs`): type=`int`, argtype=`static`.
Observation frames for prediction.
The default value is `8`.
- `--pmove`: type=`int`, argtype=`static`.
Index of the reference point when moving trajectories.
The default value is `-1`.
- `--pred_frames` (short for `-pred`): type=`int`, argtype=`static`.
Prediction frames.
The default value is `12`.
- `--protate`: type=`float`, argtype=`static`.
Reference degree when rotating trajectories.
The default value is `0.0`.
- `--pscale`: type=`str`, argtype=`static`.
Index of the reference point when scaling trajectories.
The default value is `autoref`.
- `--restore_args`: type=`str`, argtype=`temporary`.
Path to restore the reference args before training. It will not restore any args if `args.restore_args == 'null'`.
The default value is `null`.
Expand Down Expand Up @@ -321,12 +337,33 @@ About the `argtype`:
- `--feature_dim`: type=`int`, argtype=`static`.
Feature dimensions that are used in most layers.
The default value is `128`.
- `--key_points`: type=`str`, argtype=`static`.
A list of key time steps to be predicted in the agent model. For example, `'0_6_11'`.
The default value is `0_6_11`.
- `--loada` (short for `-la`): type=`str`, argtype=`temporary`.
Path to load the first-stage agent model.
The default value is `null`.
- `--loadb` (short for `-lb`): type=`str`, argtype=`temporary`.
Path to load the second-stage handler model.
The default value is `null`.
- `--msn_hotel_fix`: type=`int`, argtype=`static`.
Fix the training process of MSN on ETH-hotel by applying rotation.
The default value is `0`.
- `--preprocess`: type=`str`, argtype=`static`.
Controls whether to run any pre-process before the model inference. It accepts a 3-bit-like string value (like `'111'`): - The first bit: `MOVE` trajectories to (0, 0); - The second bit: re-`SCALE` trajectories; - The third bit: `ROTATE` trajectories.
The default value is `111`.
### First-stage silverballers args
- `--depth`: type=`int`, argtype=`static`.
Depth of the random noise vector.
The default value is `16`.
### Second-stage silverballers args
- `--points`: type=`int`, argtype=`static`.
The number of keypoints accepted in the handler model.
The default value is `1`.
<!-- DO NOT CHANGE THIS LINE -->
## Thanks
Expand Down

0 comments on commit 443caae

Please sign in to comment.