Gpen-bfr-2048.pth
# Load the model model = torch.load('gpen-bfr-2048.pth', map_location=torch.device('cpu'))