From 7773651e8ee20989e137f69d8d948adb54463127 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 24 Nov 2019 18:38:30 -1000 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ecd8f4fd..09eb70c3 100644 --- a/train.py +++ b/train.py @@ -417,7 +417,7 @@ if __name__ == '__main__': parser.add_argument('--arc', type=str, default='default', help='yolo architecture') # defaultpw, uCE, uBCE parser.add_argument('--prebias', action='store_true', help='transfer-learn yolo biases prior to training') parser.add_argument('--name', default='', help='renames results.txt to results_name.txt if supplied') - parser.add_argument('--device', default='', help='device id (i.e. 0 or 0,1) or cpu') + parser.add_argument('--device', default='', help='device id (i.e. 0 or 0,1 or cpu)') parser.add_argument('--adam', action='store_true', help='use adam optimizer') parser.add_argument('--var', type=float, help='debug variable') opt = parser.parse_args()