def initialize_states(self, states):
states.units.source = ListEntry()
states.units.target = ListEntry()
states.segments.source = ListEntry()
states.segments.target = ListEntry()
states.unit_queue.source = QueueEntry()
states.unit_queue.target = QueueEntry()
states.encoder_cache = [{} for _ in self.models]
states.decoder_cache = [{} for _ in self.models]
states.segment = False
states.encoding_time, states.decoding_time = 0, 0
def initialize_states(self, states):
self.feature_extractor.clear_cache()
states.units.source = TensorListEntry()
states.units.target = ListEntry()
states.incremental_states = dict()