On a cpu device, how to load checkpoint saved on gpu device

[Towsif_Ahamed]

torch.load(WEIGHTS_FILE, map_location=torch.device(‘cpu’) )