parameters changed for optimal values.

wwqqqqq · Mar 14, 2019 · 5dccdad · 5dccdad
1 parent 1d75e0f
commit 5dccdad
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tf_pose/train.py b/tf_pose/train.py
@@ -31,10 +31,10 @@
     parser.add_argument('--model', default='mobilenet_v2_1.4', help='model name')
     parser.add_argument('--datapath', type=str, default='/data/public/rw/coco/annotations')
     parser.add_argument('--imgpath', type=str, default='/data/public/rw/coco/')
-    parser.add_argument('--batchsize', type=int, default=96)
-    parser.add_argument('--gpus', type=int, default=1)
-    parser.add_argument('--max-epoch', type=int, default=300)
-    parser.add_argument('--lr', type=str, default='0.01')
+    parser.add_argument('--batchsize', type=int, default=64)
+    parser.add_argument('--gpus', type=int, default=4)
+    parser.add_argument('--max-epoch', type=int, default=600)
+    parser.add_argument('--lr', type=str, default='0.001')
     parser.add_argument('--tag', type=str, default='test')
     parser.add_argument('--checkpoint', type=str, default='')