This commit is contained in:
Glenn Jocher 2019-08-23 15:35:39 +02:00
parent 8ef49f2560
commit d279aa0021
1 changed files with 1 additions and 1 deletions

View File

@ -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)