updates
This commit is contained in:
parent
8ef49f2560
commit
d279aa0021
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
|
||||||
|
|
||||||
elif weights.endswith('.weights'): # darknet format
|
else: # 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