def load_model(self, epoch=None): self.label_map = pickle.load(open("%s.label_map" % self.model_name_prefix, "rb")) super(PPRelationModel, self).load_model(epoch)