right, thanks,  i was asking if there was a generalized loader that didnt require that. I was hacking it like this,
from collections import OrderedDict as OD
#...
          if device == "cuda":
                self.model = torch.nn.DataParallel(self.model)
                self.model.load_state_dict(state_dict)
            else:
                state = OD([(key.split("module.")[-1], state_dict[key]) for key in state_dict])

                self.model.load_state_dict(state)
#...