Views
No views yet
1import torch 2from lapa.models.lapa import LAPA 3 4ckpt = torch.load('lapa.pt', map_location='cpu') 5model = LAPA() 6model.load_state_dict(ckpt['model']) 7model.eval()