what.models.detection.ssd.utils.misc
1import torch 2 3 4def save_checkpoint(epoch, net_state_dict, optimizer_state_dict, best_score, checkpoint_path, model_path): 5 torch.save({ 6 'epoch': epoch, 7 'model': net_state_dict, 8 'optimizer': optimizer_state_dict, 9 'best_score': best_score 10 }, checkpoint_path) 11 torch.save(net_state_dict, model_path) 12 13 14def load_checkpoint(checkpoint_path): 15 return torch.load(checkpoint_path) 16 17 18def freeze_net_layers(net): 19 for param in net.parameters(): 20 param.requires_grad = False
def
save_checkpoint( epoch, net_state_dict, optimizer_state_dict, best_score, checkpoint_path, model_path):
def
load_checkpoint(checkpoint_path):
def
freeze_net_layers(net):