added more options to the main scripts
authorManu Mathew <a0393608@ti.com>
Thu, 14 May 2020 03:53:29 +0000 (09:23 +0530)
committerManu Mathew <a0393608@ti.com>
Thu, 14 May 2020 07:56:59 +0000 (13:26 +0530)
scripts/train_classification_main.py
scripts/train_segmentation_main.py

index 94e7524e9631bb708f3400668021ff6868740865..e7e01f1d864d4288cb27d87a875ef6c7e1aa3bde 100755 (executable)
@@ -31,6 +31,7 @@ parser.add_argument('--resume', type=str, default=None, help='resume an unfinish
 parser.add_argument('--phase', type=str, default=None, help='training/calibration/validation')
 parser.add_argument('--evaluate_start', type=str2bool, default=None, help='Whether to run validation before the training')
 parser.add_argument('--workers', type=int, default=None, help='number of workers for dataloading')
+parser.add_argument('--save_onnx', type=str2bool, default=None, help='whether to export/save onnx graph or not')
 #
 parser.add_argument('--quantize', type=str2bool, default=None, help='Quantize the model')
 parser.add_argument('--histogram_range', type=str2bool, default=None, help='run only evaluation and no training')
@@ -44,13 +45,13 @@ parser.add_argument('--freeze_bn', type=str2bool, default=None, help='freeze the
 parser.add_argument('--shuffle', type=str2bool, default=None, help='whether to shuffle the training set or not')
 parser.add_argument('--shuffle_val', type=str2bool, default=None, help='whether to shuffle the validation set or not')
 parser.add_argument('--epoch_size', type=float, default=None, help='epoch size. options are: 0, fraction or number. '
-                                                                   '0 will use th full epoch. '
-                                                                   'using a number will cause the epoch to have that many iterations'
-                                                                   'using a fraction will reduce the iterations used for one epoch to that fraction of the whole. ')
+                                                                   '0 will use the full epoch. '
+                                                                   'using a number will cause the epoch to have that many images. '
+                                                                   'using a fraction will reduce the number of images used for one epoch. ')
 parser.add_argument('--epoch_size_val', type=float, default=None, help='epoch size for validation. options are: 0, fraction or number. '
-                                                                   '0 will use th full epoch. '
-                                                                   'using a number will cause the epoch to have that many iterations. '
-                                                                   'using a fraction will reduce the iterations used for one epoch to that fraction of the whole. ')
+                                                                   '0 will use the full epoch. '
+                                                                   'using a number will cause the epoch to have that many images. '
+                                                                   'using a fraction will reduce the number of images used for one epoch. ')
 #
 cmds = parser.parse_args()
 
index 93e18fcddfddaf2fd568101f05652cc51736f452..a87dd911c1c56e35cfbdf6abc2fb98bf424639d0 100755 (executable)
@@ -30,6 +30,7 @@ parser.add_argument('--pretrained', type=str, default=None, help='pretrained mod
 parser.add_argument('--resume', type=str, default=None, help='resume an unfinished training from this model')
 parser.add_argument('--phase', type=str, default=None, help='training/calibration/validation')
 parser.add_argument('--evaluate_start', type=str2bool, default=None, help='Whether to run validation before the training')
+parser.add_argument('--workers', type=int, default=None, help='number of workers for dataloading')
 parser.add_argument('--save_onnx', type=str2bool, default=None, help='Whether to export onnx model or not')
 #
 parser.add_argument('--quantize', type=str2bool, default=None, help='Quantize the model')
@@ -44,13 +45,13 @@ parser.add_argument('--freeze_bn', type=str2bool, default=None, help='freeze the
 parser.add_argument('--shuffle', type=str2bool, default=None, help='whether to shuffle the training set or not')
 parser.add_argument('--shuffle_val', type=str2bool, default=None, help='whether to shuffle the validation set or not')
 parser.add_argument('--epoch_size', type=float, default=None, help='epoch size. options are: 0, fraction or number. '
-                                                                   '0 will use th full epoch. '
-                                                                   'using a number will cause the epoch to have that many iterations'
-                                                                   'using a fraction will reduce the iterations used for one epoch to that fraction of the whole. ')
+                                                                   '0 will use the full epoch. '
+                                                                   'using a number will cause the epoch to have that many images. '
+                                                                   'using a fraction will reduce the number of images used for one epoch. ')
 parser.add_argument('--epoch_size_val', type=float, default=None, help='epoch size for validation. options are: 0, fraction or number. '
-                                                                   '0 will use th full epoch. '
-                                                                   'using a number will cause the epoch to have that many iterations. '
-                                                                   'using a fraction will reduce the iterations used for one epoch to that fraction of the whole. ')
+                                                                   '0 will use the full epoch. '
+                                                                   'using a number will cause the epoch to have that many images. '
+                                                                   'using a fraction will reduce the number of images used for one epoch. ')
 #
 cmds = parser.parse_args()