updates
This commit is contained in:
parent
5a0575af3a
commit
cc419d88ea
2
train.py
2
train.py
|
@ -15,7 +15,7 @@ parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each i
|
||||||
parser.add_argument('-resume', default=False, help='resume training flag')
|
parser.add_argument('-resume', default=False, help='resume training flag')
|
||||||
parser.add_argument('-batch_report', default=False, help='report TP, FP, FN, P and R per batch (slower)')
|
parser.add_argument('-batch_report', default=False, help='report TP, FP, FN, P and R per batch (slower)')
|
||||||
parser.add_argument('-optimizer', default='SGD', help='optimizer')
|
parser.add_argument('-optimizer', default='SGD', help='optimizer')
|
||||||
parser.add_argument('-freeze_darknet53', default=True, help='freeze darknet53.conv.74 layers for first epoch')
|
parser.add_argument('-freeze_darknet53', default=False, help='freeze darknet53.conv.74 layers for first epoch')
|
||||||
parser.add_argument('-var', type=float, default=0, help='optional test variable')
|
parser.add_argument('-var', type=float, default=0, help='optional test variable')
|
||||||
opt = parser.parse_args()
|
opt = parser.parse_args()
|
||||||
print(opt)
|
print(opt)
|
||||||
|
|
Loading…
Reference in New Issue