Skip to content

Commit c0fd05b

Browse files
committed
Mention gradient clipping in the readme.
1 parent d0a62e9 commit c0fd05b

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,9 @@ The code should closely match the TensorFlow version (including the hyperparamet
7676
- Then, pass `--optimizer rmsprop_tf` to `main.py` to use it
7777

7878

79+
- You can turn gradient clipping off by setting `--grad_clip_off True`
80+
81+
7982
- The original training was on TPUs, this code enables only GPU and CPU training
8083
- Input data augmentation methods are the same, but due to randomness they are not applied in the same manner
8184
- Cause: Batches and images cannot be shuffled as in the original TPU training, and the augmentation seed is also different

0 commit comments

Comments
 (0)