Adapters
English
code
medical
UANN / Helper_functions.py
dnnsdunca's picture
Create Helper_functions.py
455a50b verified
raw
history blame
220 Bytes
# utils/helper_functions.py
import torch
def save_model(model, path):
torch.save(model.state_dict(), path)
def load_model(model, path):
model.load_state_dict(torch.load(path))
model.eval()
return model