|
@@ -35,6 +35,7 @@ def build_yolo_optimizer(cfg, model, resume=None):
|
|
|
optimizer.add_param_group({"params": param_dicts[2], "weight_decay": cfg.weight_decay})
|
|
optimizer.add_param_group({"params": param_dicts[2], "weight_decay": cfg.weight_decay})
|
|
|
|
|
|
|
|
start_epoch = 0
|
|
start_epoch = 0
|
|
|
|
|
+ cfg.best_map = -1.
|
|
|
if resume and resume != 'None':
|
|
if resume and resume != 'None':
|
|
|
checkpoint = torch.load(resume)
|
|
checkpoint = torch.load(resume)
|
|
|
# checkpoint state dict
|
|
# checkpoint state dict
|
|
@@ -47,8 +48,6 @@ def build_yolo_optimizer(cfg, model, resume=None):
|
|
|
print('--Load best metric from the checkpoint: ', resume)
|
|
print('--Load best metric from the checkpoint: ', resume)
|
|
|
best_map = checkpoint["mAP"]
|
|
best_map = checkpoint["mAP"]
|
|
|
cfg.best_map = best_map
|
|
cfg.best_map = best_map
|
|
|
- else:
|
|
|
|
|
- cfg.best_map = -1.
|
|
|
|
|
del checkpoint, checkpoint_state_dict
|
|
del checkpoint, checkpoint_state_dict
|
|
|
except:
|
|
except:
|
|
|
print("No optimzier in the given checkpoint.")
|
|
print("No optimzier in the given checkpoint.")
|