CUDA trained model checkpoint load in CPU

1

checkpoint = torch.load('model_best.pth.tar', map_location=lambda storage, loc: storage)

Gives me an Assertion Error, CUDA not enabled.