Loading a policy trained on the gpu on a cpu only system

[samibouziri]

Hey,
I just noticed that this line will not work if the agent is trained on the gpu and than deployed on a cpu only system.
A simple fix for this would be:
loaded_dict = torch.load(path, map_location=torch.device(self.device))