import torch


def load_model(model_path=''):
    """
    Load saved model from file
    :param model_path: mlp.pth prepared using mlp.py
    :return net: loaded model
    """
    print(f'[INFO]: Loading saved model...')
    net = torch.load(model_path)
    net = net.to('cuda:0')
    net.eval()
    return net