'Set the state dicts of the modules from saved states.'

def set_states(self, states):

    self.lt.load_state_dict(states['lt'])
    self.encoder.load_state_dict(states['encoder'])
    self.decoder.load_state_dict(states['decoder'])
    self.h2o.load_state_dict(states['h2o'])
    self.longest_label = states['longest_label']
