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
|
||||
del chkpt
|
||||
|
||||
elif weights.endswith('.weights'): # darknet format
|
||||
else: # darknet format
|
||||
# possible weights are 'yolov3.weights', 'yolov3-tiny.conv.15', 'darknet53.conv.74' etc.
|
||||
cutoff = load_darknet_weights(model, weights)
|
||||
|
||||
|
|
Loading…
Reference in New Issue