Missed clip-mode arg for repo train script

3 years ago · 01653db104
parent 4f49b94311
commit 01653db104
1 changed files with 2 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -116,7 +116,8 @@ parser.add_argument('--weight-decay', type=float, default=0.0001,
                    help='weight decay (default: 0.0001)')
 parser.add_argument('--clip-grad', type=float, default=None, metavar='NORM',
                    help='Clip gradient norm (default: None, no clipping)')
-
+parser.add_argument('--clip-mode', type=str, default='norm',
+                    help='Gradient clipping mode. One of ("norm", "value", "agc")')


 # Learning rate schedule parameters