updates
This commit is contained in:
parent
d279aa0021
commit
7593dedc4c
2
train.py
2
train.py
|
@ -121,7 +121,7 @@ def train():
|
||||||
start_epoch = chkpt['epoch'] + 1
|
start_epoch = chkpt['epoch'] + 1
|
||||||
del chkpt
|
del chkpt
|
||||||
|
|
||||||
else: # darknet format
|
elif len(weights) > 0: # darknet format
|
||||||
# possible weights are 'yolov3.weights', 'yolov3-tiny.conv.15', 'darknet53.conv.74' etc.
|
# possible weights are 'yolov3.weights', 'yolov3-tiny.conv.15', 'darknet53.conv.74' etc.
|
||||||
cutoff = load_darknet_weights(model, weights)
|
cutoff = load_darknet_weights(model, weights)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue