readme - train 50 diffusion steps

README.md

@@ -403,6 +403,7 @@ python -m train.train_mdm --save_dir save/my_name --dataset humanact12 --cond_ma
 ```
 </details>
 
+* Use `--diffusion_steps 50` to train the faster model with less diffusion steps.
 * Use `--device` to define GPU id.
 * Use `--arch` to choose one of the architectures reported in the paper `{trans_enc, trans_dec, gru}` (`trans_enc` is default).
 * Add `--train_platform_type {ClearmlPlatform, TensorboardPlatform}` to track results with either [ClearML](https://clear.ml/) or [Tensorboard](https://www.tensorflow.org/tensorboard).