def save_model()

in notebook/source/monai_dicom.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def save_model(model, model_dir):
    logger.info("Saving the model.")
    path = os.path.join(model_dir, 'model.pth')
    # recommended way from http://pytorch.org/docs/master/notes/serialization.html
    torch.save(model.cpu().state_dict(), path)