fix map_locations

README.md

@@ -12,7 +12,7 @@ The file is given as a state_dict. Thus to initialize the model run:
 
 ```
 # Load pretrained weights
-state_dict = torch.load(model_path, map_location=torch.device('cpu'))['model']
+state_dict = torch.load(model_path)['model']
 # Load torchvision model
 model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights=None)
 # Adapt input convolution