updates
This commit is contained in:
parent
a6d69cefe0
commit
c8e4a19879
2
train.py
2
train.py
|
@ -10,7 +10,7 @@ parser.add_argument('-epochs', type=int, default=100, help='number of epochs')
|
|||
parser.add_argument('-batch_size', type=int, default=16, help='size of each image batch')
|
||||
parser.add_argument('-data_config_path', type=str, default='cfg/coco.data', help='data config file path')
|
||||
parser.add_argument('-cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path')
|
||||
parser.add_argument('-img_size', type=int, default=32 * 19, help='size of each image dimension')
|
||||
parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each image dimension')
|
||||
parser.add_argument('-resume', default=False, help='resume training flag')
|
||||
opt = parser.parse_args()
|
||||
print(opt)
|
||||
|
|
Loading…
Reference in New Issue